Python 中的 map() 和 filter() 函数

在本文中，将介绍 Python 中的 map() 和 filter() 函数，两个函数在使用方法上类似。

map()

map() 会根据提供的函数对指定序列做映射。即可以将某个函数应用于可迭代对象的每个元素（如列表或元组），并返回一个新的修改后的可迭代对象。

map() 函数语法格式：

map(function, iterable, …)

参数：

function – 函数，除了单独定义函数外，还可以使用 lambda 表达式。

iterable – 一个或多个序列。

返回值：

Python 2.x 返回列表。

Python 3.x 返回迭代器。

示例1：计算列表元素的平方

def square(a):
    return a**2
lst1 = [1, 2, 3, 4, 5, 6]
lst2 = list(map(square,lst1))
print(lst2) # 输出：[1, 4, 9, 16, 25, 36]

以上示例中，单独定义了一个函数（square），也可以使用 Lambda 函数来使代码更具可读性。

lst1 = [1, 2, 3, 4, 5, 6]
lst2 = list(map(lambda a: a**2, lst1))
print(lst2)

示例2：转换列表元素为大写

使用内置字符串方法 str.upper() 结合 map() 函数来实现。

str1 = ["Hello", "world"]
str2 = list(map(str.upper, str1))
print(str2)# 输出：['HELLO', 'WORLD']

示例3：计算两个列表元素之和

map()函数也可以与多个可迭代对象一起使用。

lst1 = [1, 2, 3, 4, 5, 6]
lst2 = [10, 20, 30, 40, 50, 60]
sums = list(map(lambda x, y: x + y, lst1, lst2))
print(sums) # 输出：[11, 22, 33, 44, 55, 66]

filter()

filter() 函数用于过滤序列，过滤掉不符合条件的元素，返回由符合条件元素组成的新列表。用于根据特定条件（或函数）从可迭代对象（如列表或元组）中过滤掉元素。

filter() 函数语法格式:

filter(function, iterable)

参数：

function – 判断函数。从可迭代对象中获取一个元素并返回 True 或 False。然后，filter() 函数返回一个迭代器，该迭代器仅包含该函数返回 True 的元素。

iterable – 可迭代对象。

返回值：

返回列表。

示例1：从列表中筛选出偶数

lst1 = [1, 2, 3, 4, 5, 6]
lst2 = list(filter(lambda a: a%2 == 0, lst1))
print(lst2) # 输出：[2, 4, 6]

示例2：从列表筛选数值

lst1 = [1, 2, 3, "Hello", "world"]
lst2 = list(filter(lambda a: type(a) == int, lst1))
print(lst2) # 输出：[1, 2, 3]

Python 中的 map()、filter()函数都可以将某个函数应用于可迭代对象的每个元素，。不过，map()函数是修改元素，filter()函数是过滤元素。

文章创作不易，如果您喜欢这篇文章，请关注、点赞并分享给朋友。如有意见和建议，请在评论中反馈。