目录
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表达式清晰简洁表达。