假设我们有一个函数add如下所示
def add(x, y):
return x + y
我们想为数组应用map函数
map(add, [1, 2, 3], 2)
语义是我想给数组的每个元素加2。但是map函数在第三个参数中也需要一个列表。
注意:为了简单起见,我使用了添加示例。原来的函数要复杂得多。当然,在add函数中设置y的默认值是不可能的,因为每次调用都会改变它。
假设我们有一个函数add如下所示
def add(x, y):
return x + y
我们想为数组应用map函数
map(add, [1, 2, 3], 2)
语义是我想给数组的每个元素加2。但是map函数在第三个参数中也需要一个列表。
注意:为了简单起见,我使用了添加示例。原来的函数要复杂得多。当然,在add函数中设置y的默认值是不可能的,因为每次调用都会改变它。
当前回答
#multi argument
def joke(r):
if len(r)==2:
x, y = r
return x + y
elif len(r)==3:
x,y,z=r
return x+y+z
#using map
print(list(map(joke,[[2,3],[3,4,5]])))
输出= [6,12]
如果像上面的情况,只是想使用功能
def add(x,y):
ar =[]
for xx in x:
ar.append(xx+y)
return ar
print(list(map(add,[[3,2,4]],[2]))[0])
output = [5,4,6]
注:可以根据需要修改。
其他回答
如果你真的需要使用map函数(就像我的课堂作业…),你可以使用一个包装器函数只有一个参数,其余的传递给原始的一个在它的主体;例如:
extraArguments = value
def myFunc(arg):
# call the target function
return Func(arg, extraArguments)
map(myFunc, itterable)
又脏又丑,还是有用的
一个选项是列表推导式:
[add(x, 2) for x in [1, 2, 3]]
更多的选择:
a = [1, 2, 3]
import functools
map(functools.partial(add, y=2), a)
import itertools
map(add, a, itertools.repeat(2, len(a)))
In:nums = [1,2,3]
在:文件夹(添加,nums, [2]* en(nums))
出:[3,4,5]
你可以将lambda和map一起包含:
list(map(lambda a: a+2, [1, 2, 3]))
我相信星图就是你所需要的:
from itertools import starmap
def test(x, y, z):
return x + y + z
list(starmap(test, [(1, 2, 3), (4, 5, 6)]))