Python: Split values into two groups, based on the result of the given filter list

Python List: Exercise - 231 with Solution

Write a Python program to split values into two groups, based on the result of the given filter list.

• Use a list comprehension and zip() to add elements to groups, based on filter.
• If filter has a truthy value for any element, add it to the first group, otherwise add it to the second group.

Sample Solution:

Python Code:

def bifurcate(colors, filter):
  return [
    [x for x, flag in zip(colors, filter) if flag],
    [x for x, flag in zip(colors, filter) if not flag]
  ]
print(bifurcate(['red', 'green', 'blue', 'pink'], [True, True, False, True]))

Sample Output:

[['red', 'green', 'pink'], ['blue']]

Flowchart:

Visualize Python code execution:

The following tool visualize what the computer is doing step-by-step as it executes the said program:

Python Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a Python program to find the indexes of all elements in the given list that satisfy the provided testing function.
Next: Write a Python program to chunk a given list into smaller lists of a specified size.