假设我们有一个函数add如下所示

def add(x, y):
    return x + y

我们想为数组应用map函数

map(add, [1, 2, 3], 2)

语义是我想给数组的每个元素加2。但是map函数在第三个参数中也需要一个列表。

注意:为了简单起见,我使用了添加示例。原来的函数要复杂得多。当然,在add函数中设置y的默认值是不可能的,因为每次调用都会改变它。


当前回答

文档明确建议这是itertools.repeat的主要用途:

创建一个反复返回object的迭代器。除非指定了times参数,否则无限期地运行。用作map()的参数,用于指定被调用函数的不变形参。也与zip()一起用于创建元组记录的不变部分。

并且没有理由将len([1,2,3])作为times参数;Map在第一个迭代对象被消耗后立即停止,所以无限迭代对象是完全没问题的:

>>> from operator import add
>>> from itertools import repeat
>>> list(map(add, [1,2,3], repeat(4)))
[5, 6, 7]

事实上,这相当于文档中repeat的例子:

>>> list(map(pow, range(10), repeat(2)))
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

这是一个很好的惰性函数语言解决方案,在python迭代器术语中也是完全可读的。

其他回答

Map可以包含多个参数,标准方式是

map(add, a, b)

在你的问题中,应该是

map(add, a, [2]*len(a))
#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(add, [(x, 2) for x in [1,2,3]])

并将add的实现更改为接受一个元组,即

def add(t):
   x, y = t
   return x+y

这可以处理任何添加参数都是动态的复杂用例。

In:nums = [1,2,3]

在:文件夹(添加,nums, [2]* en(nums))

出:[3,4,5]

如果您有它可用,我会考虑使用numpy。对于这些类型的操作,它非常快:

>>> import numpy
>>> numpy.array([1,2,3]) + 2
array([3, 4, 5])

这是假设您的实际应用程序正在执行数学操作(可以向量化)。