简介
map()是 Python 内置的高阶函数,它接收一个函数 func 和一个 list,并通过把函数 func依次作用在 list 的每个元素上,得到一个新的 list 并返回。
一、当list只有一个时
当list只有一个时,将函数func作用于这个list的每个元素上,并返回一个map对象。
def func(x):
return x**2
print(map(func, [1,2,3,4,5]))
for x in map(func, [1,2,3,4,5])):
print(x)
输出
< map object at 0x0000029C8FD08940 >
1
4
9
16
25
另一个应用案例:将字符串列表中每个字母的首字母变为大写,其余
def func(str):
stmp = str[0:1].upper() + str[1:].lower()
return stmp
lst = ['aWm', 'AKm', 'kar', 'groza']
print(list(map(func, lst)))
输出结果:
['Awm', 'Akm', 'Kar', 'Groza']
二、当list有多个时
当存在多个list对象时,map并行地处理传入的list,调用满足下图规则:
def func(x,y,z):
result1 = x**y**z
result2 = x*y*z
return (result1, result2)
print(list(map(func, [1,2,3],[1,1,1],[3,2,1])))
输出结果:
>>> [(1, 3), (2, 4), (3, 3)]
一些需要注意的细节
在python3中map可以处理长度不同的list(但是列表成员类型必须相同),他会将所有列表截断为最小长度
for i in map(lambda x,y:(x**y,x+y),[1,2,3,4,5,6],[1,2]):
print(i)
输出结果:
>>> (1, 2)
(4, 4)
一些特殊用法
1、用作类型转换
for x in map(int, '1234'):
print(x)
输出结果:
>>> 1
2
3
4
2、用作zip函数
如果函数是 None,自动假定一个‘identity’函数,这时候就是模仿 zip()
在python2.x中有
print map(None, [1,2,3],[3,2,1])
运行结果:
>>> [(1, 3), (2, 2), (3, 1)]
但是在 python3中,返回是一个迭代器,所以它其实是不可调用的