map()
函数详解
1. 关于map()
函数的描述:
map()
函数会根据你提供的函数对指定序列中的元素做映射。更具体点来说就是:第一个参数 function
(也就是你想实现功能的函数) 以(第二个、第三个、…参数序列)iterable
中的每一个元素调用 function
函数,返回包含每次 function 函数返回值的新列表。
2. 语法:
map(function, iterable, ...)
function
– 函数iterable
– 一个或多个序列- 返回值为迭代器对象
iterable
3. 代码示例:
- 基本用法:
>>> def square(x) : # 计算平方数
... return x ** 2
...
>>> map(square, [1,2,3,4,5]) # 计算列表各个元素的平方
<map object at 0x100d3d550> # 返回迭代器
>>> list(map(square, [1,2,3,4,5])) # 使用 list() 转换为列表
[1, 4, 9, 16, 25]
>>> list(map(lambda x: x ** 2, [1, 2, 3, 4, 5])) # 使用 lambda 匿名函数
[1, 4, 9, 16, 25]
>>>
- 进阶用法:
如果 zip(*array) 不经过 x,y,z=zip(*array); 拆分成 x,y,z 三个变量,那么 [[1,4,7],[2,5,8],[3,6,9]]; 被 zip(*array) 之后的结果恰好是 [(1, 2, 3), (4, 5, 6), (7, 8, 9)],刚好形成一个转置的关系。这对于所有 array=[[1,2,3],[4,5,6],[7,8,9]]; 的二维数组都是一样的。当然 [(1, 2, 3), (4, 5, 6), (7, 8, 9)] 还不是我们需要的最后的结果。因为只是一个存放 tuple 的 list,我们要保持原来 list 是存 list 的一致性,所以要应用到上方的 map 函数。
因此对于一个数组的转置,代码如下:
>>> array = [[1, 4], [2, 5], [3, 6]];
>>> map(list, zip(*array));
[[1, 2, 3], [4, 5, 6]]
其他实例:
>>> list(zip([1, 3, 5, 7, 9], [2, 4, 6, 8, 10]))
[(1, 2), (3, 4), (5, 6), (7, 8), (9, 10)]
>>> list(map(lambda x,y : [x,y], [1,3,5,7,9], [2,4,6,8,10]))
[[1, 2], [3, 4], [5, 6], [7, 8], [9, 10]]
4.注意:
filter
是通过生成True
和False
组成的迭代器将可迭代对象中不符合条件的元素过滤掉;而map
返回的则是True
和False
组成的迭代器。
>>> res1 = map(lambda n: n > 5, range(10))
>>> lt1 = list(res1)
>>> print(lt1)
[False, False, False, False, False, False, True, True, True, True]
>>> res2 = filter(lambda n: n > 5, range(10))
>>> lt = list(res2)
>>> print(lt)
[6, 7, 8, 9]
- 易错:以下使用会报错。不要理所当然觉得
zip
会自动帮你将[(1,2),(3,4),(5,6),(7,8),(9,10)]
解包赋值给x=1或3;5;7;9
y=2或4;6;8;10
a = list(map(lambda x, y: x+y, zip([1, 3, 5, 7, 9], [2, 4, 6, 8, 10])))
print(a)
# 运行结果
a = list(map(lambda x, y: x+y, zip([1, 3, 5, 7, 9], [2, 4, 6, 8, 10])))
TypeError: <lambda>() missing 1 required positional argument: 'y'