How do I use itertools.filter in Python?

How do I use itertools.filter in Python?

You can use itertools.filter to filter elements from an iterable based on a function that returns True or False.

Can you give me an example of how to use it?

Sure! Here's a simple example:

import itertools

# A function that returns True for even numbers
def is_even(n):
    return n % 2 == 0

# Using itertools.filter to filter even numbers from a list
numbers = [1, 2, 3, 4, 5, 6]
filtered_numbers = itertools.filter(is_even, numbers)
print(list(filtered_numbers))  # Output: [2, 4, 6]

What if I want to filter based on a condition instead of a function?

You can use a lambda function for that! For example:

filtered_numbers = itertools.filter(lambda x: x > 3, numbers)
print(list(filtered_numbers))  # Output: [4, 5, 6]