filter() in python
Last Updated :
20 Jun, 2024
The filter() method filters the given sequence with the help of a function that tests each element in the sequence to be true or not.
Python filter() Syntax
The filter() method in Python has the following syntax:
Syntax: filter(function, sequence)
Parameters:
- function: function that tests if each element of a sequence is true or not.
- sequence: sequence which needs to be filtered, it can be sets, lists, tuples, or containers of any iterators.
Returns: an iterator that is already filtered.
Python filter Function Examples
Let us see a few examples of the filter() function in Python.
Python Filter Function with a Custom Function
In this example, we are using the filter function along with a custom function “fun()” to filter out vowels from the Python List.
Python
# function that filters vowels
def fun(variable):
letters = ['a', 'e', 'i', 'o', 'u']
if (variable in letters):
return True
else:
return False
# sequence
sequence = ['g', 'e', 'e', 'j', 'k', 's', 'p', 'r']
# using filter function
filtered = filter(fun, sequence)
print('The filtered letters are:')
for s in filtered:
print(s)
Output:
The filtered letters are:
e
e
Filter Function in Python with Lambda
Python filter() function is normally used with Lambda functions. In this example, we are using the lambda function to filter out the odd and even numbers from a list.
Python
# a list contains both even and odd numbers.
seq = [0, 1, 2, 3, 5, 8, 13]
# result contains odd numbers of the list
result = filter(lambda x: x % 2 != 0, seq)
print(list(result))
# result contains even numbers of the list
result = filter(lambda x: x % 2 == 0, seq)
print(list(result))
Output:
[1, 3, 5, 13]
[0, 2, 8]
Filter Function in Python with Lambda and Custom Function
In this program, we will use both a custom function “is_multiple_of_3()” as well as a lambda function. The filter() function is used to apply this function to each element of the numbers list, and the lambda function is used to iterate over each element of the list before applying the condition. This way, we can perform additional operations on each element before applying the condition.
Python
# Define a function to check
# if a number is a multiple of 3
def is_multiple_of_3(num):
return num % 3 == 0
# Create a list of numbers to filter
numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
# Use filter and a lambda function to
# filter the list of numbers and only
# keep the ones that are multiples of 3
result = list(filter(lambda x: is_multiple_of_3(x), numbers))
# Print the result
print(result)
Time complexity analysis
- The filter function is used to filter the list of numbers, and it applies the lambda function to each element of the list. The time complexity of the filter function is O(n), where n is the number of elements in the list.
- The time complexity of the lambda function is constant, O(1), since it only performs a single arithmetic operation. Therefore, the overall time complexity of the program is O(n).
Auxiliary Space analysis
The program uses a list to store the filtered numbers, so the space complexity is proportional to the number of filtered numbers. In the worst case, if all numbers are multiples of 3, the filtered list will have n/3 elements. Therefore, the space complexity is O(n/3), which simplifies to O(n) in big O notation.
filter() in python – FAQs
What does the filter() function do?
The filter() function in Python filters elements from an iterable (like a list) based on a function (or None for truthy values). It returns an iterator that yields those elements for which the function returns True.
How does reduce() and filter() work in Python?
- filter(): Filters elements from an iterable based on a function.
- reduce(): Applies a function to the items of an iterable, cumulatively, from left to right, so as to reduce the iterable to a single value.
Both functions are part of Python’s functools module (Python 3.x) and can be used together to filter and then reduce data if needed.
How to filter a list in Python?
To filter a list in Python, you can use a list comprehension or the filter() function along with a lambda function or a defined function.
Using list comprehension:
# Example using list comprehension
numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9]
filtered_numbers = [num for num in numbers if num % 2 == 0]
print(filtered_numbers) # Output: [2, 4, 6, 8]
Using filter() with a lambda function:
# Example using filter() with lambda function
numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9]
filtered_numbers = list(filter(lambda x: x % 2 == 0, numbers))
print(filtered_numbers) # Output: [2, 4, 6, 8]
How to filter an object in Python?
Filtering an object in Python typically involves filtering its attributes or properties based on certain conditions. Here’s an example using filter() and a lambda function to filter a list of objects based on an attribute:
# Example filtering objects based on attribute
class Person:
def __init__(self, name, age):
self.name = name
self.age = age
people = [
Person('Alice', 25),
Person('Bob', 30),
Person('Charlie', 22)
]
# Filter people older than 25
filtered_people = list(filter(lambda person: person.age > 25, people))
for person in filtered_people:
print(person.name, person.age)
How to filter a text in Python?
Filtering text in Python usually involves searching for specific patterns or conditions within strings. Here’s an example using list comprehension to filter out words containing a specific substring:
# Example filtering text using list comprehension
text = "Python is awesome and versatile"
filtered_words = [word for word in text.split() if 'o' in word]
print(filtered_words) # Output: ['Python', 'awesome']
Please Login to comment...