map()是Python中的内置函数,可将函数应用于指定可迭代对象中的所有元素。 它使您无需使用循环即可编写简单干净的代码。

Python map()函数

map()函数采用以下形式:

map(function, iterable, ...)

它接受两个强制性参数:

  • function-为iterable的每个元素调用的函数。
  • iterable -一个或多个支持迭代的对象。 Python中的大多数内置对象,例如列表,字典和元组都是可迭代的。

在Python 3中,map()返回长度大小等于传递的可迭代对象的map对象。 在python 2中,该函数返回一个列表。

让我们看一个例子,以更好地说明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()function函数,代码将更加简单和灵活。

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()构造函数将返回的map对象转换为列表

['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()与多个可迭代项一起使用

您可以根据需要将尽可能多的可迭代对象传递给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()函数采用一个可迭代的对象以及一个函数,并将该函数应用于可迭代的每个元素。

如果您有任何问题或反馈,请随时发表评论。