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

Last update on April 19 2025 12:57:57 (UTC/GMT +8 hours)

Split Values by Filter List

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:

# Define a function 'bifurcate' that takes a list 'colors' and a list of filters 'filter' as input.
def bifurcate(colors, filter):
    # Use list comprehensions to create two lists: one with elements corresponding to True filter values and one with elements corresponding to False filter values.
    return [
        [x for x, flag in zip(colors, filter) if flag],
        [x for x, flag in zip(colors, filter) if not flag]
    ]

# Call the 'bifurcate' function with example lists 'colors' and 'filter'.
print(bifurcate(['red', 'green', 'blue', 'pink'], [True, True, False, True]))

Sample Output:

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

Flowchart:

For more Practice: Solve these Related Problems:

Write a Python program to partition a list of colors into two groups based on whether they start with a vowel or a consonant.
Write a Python program to split a list into two groups: one with elements containing a specified substring and one without.
Write a Python program to divide a list of numbers into two groups based on whether they exceed a given threshold.
Write a Python program to split a list into two groups using a filter list that categorizes elements by length.

Go to:

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.

Python Code Editor: