欢迎访问宙启技术站
智能推送

Python的filter函数如何从列表中过滤出特定的元素

发布时间:2023-07-02 09:35:47

Python中的filter()函数可以用于从列表中筛选出满足特定条件的元素。它接受两个参数:一个是函数,一个是可迭代对象。

filter()函数会遍历可迭代对象中的每个元素,并将其传递给函数进行判断。如果函数返回True,则该元素会被保留下来,否则会被过滤掉。最终,filter()函数会返回一个可迭代对象,其中包含所有满足条件的元素。下面是一些filter()函数的使用示例。

示例1:从列表中筛选出所有的偶数

numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

def is_even(num):
    return num % 2 == 0

even_numbers = list(filter(is_even, numbers))
print(even_numbers)

输出:

[2, 4, 6, 8, 10]

在上面的示例中,我们定义了一个函数is_even(),用于判断一个数是否为偶数。然后我们使用filter()函数将numbers列表中的每个元素都传递给is_even()函数进行判断,最终得到一个包含所有偶数的列表even_numbers。

示例2:从字符串列表中筛选出长度大于等于5的字符串

strings = ['hello', 'python', 'filter', 'function', 'list']

def longer_than_five(string):
    return len(string) >= 5

long_strings = list(filter(longer_than_five, strings))
print(long_strings)

输出:

['hello', 'python', 'filter', 'function']

在这个示例中,我们定义了一个函数longer_than_five(),用于判断一个字符串的长度是否大于等于5。然后我们使用filter()函数将strings列表中的每个字符串都传递给longer_than_five()函数进行判断,最终得到一个包含所有长度大于等于5的字符串的列表long_strings。

示例3:从字典列表中筛选出所有年龄大于等于18的人的姓名

people = [{'name': 'Alice', 'age': 20},
          {'name': 'Bob', 'age': 17},
          {'name': 'Charlie', 'age': 25},
          {'name': 'David', 'age': 19}]

def adult(person):
    return person['age'] >= 18

adult_names = list(map(lambda x: x['name'], filter(adult, people)))
print(adult_names)

输出:

['Alice', 'Charlie', 'David']

在这个示例中,我们定义了一个函数adult(),用于判断一个人的年龄是否大于等于18。然后我们使用filter()函数将people列表中的每个字典都传递给adult()函数进行判断,最终得到一个包含所有年龄大于等于18的人的字典列表。接着我们使用map()函数和lambda表达式提取出字典中的姓名,最终得到一个包含所有年龄大于等于18的人的姓名的列表adult_names。

通过上述示例,你可以看到filter()函数的用法非常灵活,你可以根据自己的需要编写不同的条件函数来筛选出不同类型的元素。同时,使用filter()函数可以避免显式的循环遍历列表,并且代码更为简洁高效。