【python基础】python map的用法

map的使用形式是：map(function, [x1, x2, x3 ...])，其效果是将参数列表中的每个元素，依次传入到 function 中并收集每次 function 返回的结果，然后把所有结果打包成一个 map 对象并返回。例如：

def square(x):  # 计算平方数
    return x ** 2

results = map(square, [1,2,3,4,5])  # 计算列表各个元素的平方
print(list(results))  # [1, 4, 9, 16, 25] (转化成list才能打印出来) 

可以传入多个参数列表：

def mytimes(x, y):  # 计算两个数的乘积
    return x * y

results = map(mytimes, [1, 2, 3, 4, 5], [2, 2, 2, 2, 2])
print(list(results))  # [2, 4, 6, 8, 10] (转化成list才能打印出来)

如果 function 的返回值有多个呢？

def func2(x1, x2):
    return x1, x2

map_results = map(func2, [0, 1, 2, 3, 4], [5, 6, 7, 8, 9])
print(*map_results)  # (0, 5) (1, 6) (2, 7) (3, 8) (4, 9)

使用 * 解包（参考：python 解包运算符），可以发现 map_result 中的每个元素应该都是一个 map 对象，所以我们可以把 *map_results 打包成一个列表（或者元组），即 [*map_results]，然后再调用 map，对每个 map 元素使用 List，把每个元素都转换为列表：

# map(list, [*map_results])，将 map_results 中的每个元素变成列表
print(list(map(list, [*map_results])))  # 由于map返回的最外层仍然是map，需要再套一个list来得到列表

输出：

[[0, 5], [1, 6], [2, 7], [3, 8], [4, 9]]

拓展一下：

map_results1, map_results2 = tuple(map(list, zip(*map_results)))  
print(map_results1, map_results2)

输出：

[0, 1, 2, 3, 4] [5, 6, 7, 8, 9]

这里使用了 python zip() 将(0, 5) (1, 6) (2, 7) (3, 8) (4, 9) 5个 map 对象打包成 (0, 1, 2, 3, 4), (5, 6, 7, 8, 9) 两个 map 对象，然后仍然调用 map(list, …) 来将 map 对象转换为 list，最后再将转换后的两个 list 打包成一个元组，并用 map_results1 和 map_results2 接收。