map()是Python中的内置函数,可将函数应用于给定可迭代对象中的所有元素,它使您无需使用循环即可编写简单干净的代码。
Python map()函数
map()函数采用以下形式:
map(function, iterable, ...)
它接受两个强制性参数:
1、function-为iterable的每个元素调用的函数。
2、iterable-支持迭代的一个或多个对象,Python中的大多数内置对象(如列表、字典和元组)都是可迭代的。
在Python 3中,map()返回大小等于传递的可迭代对象的map对象,在python 2中,该函数返回一个列表。参考在CentOS 8上安装Python 3和Python 2,及设置默认Python版本。
让我们看一个示例,以更好地解释map()函数的工作方式,假设我们有一个字符串列表,我们希望将列表中的每个元素都转换为大写。
一种方法是使用传统的for循环:
directions = ["north", "east", "south", "west"]
directions_upper = []
for direction in directions:
d = direction.upper()
directions_upper.append(d)
print(directions_upper)
返回:
['NORTH', 'EAST', 'SOUTH', 'WEST']
使用map()函数,代码将更加简单和灵活:
def to_upper_case(s):
return s.upper()
directions = ["north", "east", "south", "west"]
directions_upper = map(to_upper_case, directions)
print(list(directions_upper))
我们正在使用list()函数将返回的地图对象转换为列表:
['NORTH', 'EAST', 'SOUTH', 'WEST']
如果回调函数很简单,那么更多的Python方式是使用lambda函数:
directions = ["north", "east", "south", "west"]
directions_upper = map(lambda s: s.upper(), directions)
print(list(directions_upper))
注:Lambda函数是一个小的匿名函数。
这是另一个示例,显示了如何创建从1到10的平方数的列表:
squares = map(lambda n: n*n , range(1, 11))
print(list(squares))
返回:
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
注:range()函数生成一个整数序列。
将map()与多个Iterables一起使用
您可以根据需要将尽可能多的可迭代对象传递给map()函数,回调函数接受的必需输入参数的数量必须与可迭代的数量相同。
下面的示例显示如何在两个列表上执行逐元素乘法:
def multiply(x, y):
return x * y
a = [1, 4, 6]
b = [2, 3, 5]
result = map(multiply, a, b)
print(list(result))
返回:
[2, 12, 30]
相同的代码,但使用lambda函数,如下:
a = [1, 4, 6]
b = [2, 3, 5]
result = map(lambda x, y: x*y, a, b)
print(list(result))
当提供多个可迭代时,返回的对象的大小等于最短的可迭代。
让我们看一个示例,其中可迭代项的长度不同:
a = [1, 4, 6]
b = [2, 3, 5, 7, 8]
result = map(lambda x, y: x*y, a, b)
print(list(result))
多余的元素(7和8)将被忽略:
[2, 12, 30]
结论
Python的map()函数采用一个可迭代的对象以及一个函数,并将该函数应用于可迭代的每个元素。
相关主题