Python内建函数map

简介

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中，返回是一个迭代器，所以它其实是不可调用的