Python中的“filter()”函数：筛选列表、元组和字典

Python中的“filter()”函数是一种内置的高阶函数，它可以对给定的序列（列表、元组、迭代器等）进行筛选操作。筛选的结果是符合特定条件的元素组成的一个新序列。在本文中，我们将探讨如何使用“filter()”函数来筛选列表、元组和字典。

使用“filter()”函数筛选列表

“filter()”函数的第一个参数是一个函数，该函数接受一个参数并返回一个布尔值（True或False）。第二个参数是一个序列，可以是任何可迭代对象，如列表、元组、字典、字符串等。当我们调用“filter()”函数时，它会自动迭代每个元素，并将其传递给第一个参数中的函数进行测试。如果元素返回True，则将其添加到新的序列中。

例如，假设我们有一个列表，其中包含一些数字，我们希望只保留大于等于5的元素。我们可以使用以下代码：

original_list = [1, 2, 3, 5, 7, 9, 10, 11]
new_list = list(filter(lambda x: x >= 5, original_list))
print(new_list)  # Output: [5, 7, 9, 10, 11]

在这个例子中，我们使用了一个匿名函数（lambda表达式）来测试每个数字是否大于等于5。我们将原始列表作为第二个参数传递给“filter()”函数，并将返回结果转换成列表。

使用“filter()”函数筛选元组

与列表类似，我们可以使用“filter()”函数来筛选元组。唯一不同之处是“filter()”函数返回的是一个迭代器，而不是一个列表。这意味着我们需要使用“for”循环来遍历返回的结果。

例如，假设我们有一个元组，其中包含一些字符串，我们希望只保留长度大于等于5的字符串。我们可以使用以下代码：

original_tuple = ('apple', 'orange', 'banana', 'grape', 'mango', 'pear')
new_tuple = filter(lambda x: len(x) >= 5, original_tuple)
for item in new_tuple:
    print(item)

在这个例子中，我们使用一个匿名函数来测试每个字符串的长度是否大于等于5。我们将原始元组作为第二个参数传递给“filter()”函数，并使用“for”循环遍历返回的迭代器。

使用“filter()”函数筛选字典

虽然字典是无序的键值对集合，但我们可以使用“filter()”函数来筛选字典中的键或值。

如果要筛选字典中的键，我们可以使用字典的“keys()”方法来创建一个由键组成的序列，然后对其进行筛选操作。

例如，假设我们有一个字典，存储了一些城市的人口数据，我们希望只保留人口数量大于等于100万的城市。我们可以使用以下代码：

population = {'New York': 8336817, 'Los Angeles': 3979576, 'Chicago': 2693976, 'Houston': 2320268, 'Phoenix': 1680992, 'Philadelphia': 1584064, 'San Antonio': 1547253}

new_population = dict(filter(lambda x: x[1] >= 1000000, population.items()))
print(new_population)  # Output: {'New York': 8336817, 'Los Angeles': 3979576, 'Chicago': 2693976, 'Houston': 2320268, 'Phoenix': 1680992, 'Philadelphia': 1584064, 'San Antonio': 1547253}

在这个例子中，我们使用“items()”方法将字典的键值对转换成由元组组成的列表。我们使用一个匿名函数来测试每个元组的第二个元素（即人口数量）是否大于等于100万。我们将过滤后的元组列表转换成字典，并将其存储在新的变量中。

如果要筛选字典中的值，则可以使用字典的“values()”方法来创建一个由值组成的序列，并像前面一样进行筛选操作。

总结

使用“filter()”函数可以轻松地对列表、元组和字典进行筛选操作，只需提供一个测试条件函数，即可获得所需的输出。请记住，对于列表和元组，结果将返回一个新的序列；而对于字典，结果将返回一个新的字典。