目录
描述
map() 会根据提供的函数对指定序列做映射。
第一个参数 function 以参数序列中的每一个元素调用 function 函数,返回包含每次 function 函数返回值的新列表。
语法
map() 函数语法:
map(function, iterable, ...)
参数
- function -- 函数
- iterable -- 一个或多个序列
返回值
Python 2.x 返回列表。
Python 3.x 返回迭代器。
示例
>>>def square(x) : # 计算平方数
... return x ** 2
...
>>> map(square, [1,2,3,4,5]) # 计算列表各个元素的平方
[1, 4, 9, 16, 25]
>>> map(lambda x: x ** 2, [1, 2, 3, 4, 5]) # 使用 lambda 匿名函数
[1, 4, 9, 16, 25]
# 提供了两个列表,对相同位置的列表数据进行相加
>>> map(lambda x, y: x + y, [1, 3, 5, 7, 9], [2, 4, 6, 8, 10])
[3, 7, 11, 15, 19]
#如果函数有多个参数,但每个参数的序列元素个数不一样,会根据最少元素的序列进行:
>>> listx = [1,2,3,4,5,6,7] # 7 个元素
>>> listy = [2,3,4,5,6,7] # 6 个元素
>>> listz = [100,100,100,100] # 4 个元素
>>> list_result = map(lambda x,y,z : x**2 + y + z,listx, listy, listz)
>>> print(list(list_result))
[103, 107, 113, 121]
#二维数组的转置
>>> arr = [[1, 4], [2, 5], [3, 6]]
>>> list(map(list, zip(*arr)))
[[1, 2, 3], [4, 5, 6]]
#当然转置也可以使用numpy实现
>>> import numpy as np
>>> arr = [[1, 4], [2, 5], [3, 6]]
>>> arr=np.array(arr).transpose()
>>> arr
#python3下函数返回的是迭代器,不再是列表
>>> arr = [[1, 4], [2, 5], [3, 6]]
>>> for item in map(list, zip(*arr)):
... print(item)
>>> print(map(list, zip(*arr)))
<map object at 0x000001E3734E20B8>