Python中内置函数map（）用法

目录

一、案例一

二、案例二

三、案例三

四、总结

map(function, iterable, ...)

它需要两个必须的参数：

• function - 针对每一个迭代调用的函数
• iterable - 支持迭代的一个或者多个对象。在 Python 中大部分内建对象，例如 lists, dictionaries, 和 tuples 都是可迭代的。
• 如果函数有两个形参，则表示map中有两个可迭代对象，返回值有多个，接收的map结果会以列表套元组方式返回。

        list(map(lambda x,y,z:(x+y+z,x*y),a,b,c))     函数有三个形参，则需要三个迭代对象abc，返回值有两个，最终结果为：[(1,2),(1,2),(1,2)...(1,2)]

一、案例一

#自定义一个函数a,返回x*2的值
def a(x):
    return x*2
 
#定义列表
lis1=[1,3,5,7,9]
 
#对列表中的每个数运用函数a，返回迭代器
lis1_a=map(a,lis1)
 
#输出迭代器中的值
for num in lis1_a:
    print(num)
    
#输出结果
'''
2
6
10
14
18
'''

二、案例二

多个可迭代对象传递给map（）函数，然后map（）函数将这2个迭代对象传入自定义函数a（）。注意：这个自定义函数a（）的参数个数，要与传入的可迭代对象数量一致。

# 定义一个函数
# 该函数采用2参数
def a(x,y):
 
    return x * y
 
# 定义列表1
lis1 = [1, 3, 5, 7, 9]
# 定义列表2
lis2 = [2, 4, 6, 8, 10]
 
# 将两个列表中的值，传入函数a，并返回可迭代器
lis_a = map(a, lis1, lis2)
 
#输出迭代器中的值
for num in lis_a:
    print(num)
    
#输出结果
'''
2
12
30
56
90
'''

案例分析

函数a（）是我们自定义的一个函数，因为后面要对传入的2个可迭代对象进行运算。所以，函数a（）必须有2个形参。lis1和lis2是我们定义的2个列表，map（）函数将这2个列表中的中，分别传入函数a（）进行运算，返回一个迭代器lis_a。此时的lis_a的数据类型是：<class 'map'>，我们运用for循环遍历输出迭代器lis_a的值。

三、案例三

这个自定义函数a（）的参数个数，要与传入的可迭代对象数量一致。每一个函数参数都对应相应可迭代对象的每一个元素。如案例2中的函数有两个参数x,y。可迭代对象也有两个。则x表示list1的每一个元素，y表示list2的每一个元素。因此可以这样操作：

a= [1,2,3]
b= [4,5,6]
c = list(map(lambda x,y:(x+y,x*y),a,b))
for b in c:
    print(b)

“”“
输出结果：
(5, 4)
(7, 10)
(9, 18)
”“”

a= [1,2,3]
b= [4,5,6]
c = list(map(lambda x,y:(x+y,x*y,y**2),a,b))
for b in c:
    print(b)
“”“
输出结果：
(5, 4, 16)
(7, 10, 25)
(9, 18, 36)
“”“

四、总结

在炼丹中很多时候需要同时批量处理tensor，可以考虑使用map函数，lambda表达式清晰简洁表达。