在参数列表上映射函数列表(Python)

时间:2016-10-29 01:13:06

标签: python functional-programming cartesian-product

我有一个函数列表:listFunc = [g1,g2,g3]。此列表使用以下代码生成:

def g(y): 
    def f(x):
        return x+y;
    return f; 
listFunc=list(map(g, [1, 2, 3])); 

现在,我有一个参数列表ListArg = [4,5,6];

我如何仅使用[g1(4), g1(5), g1(6), g2(4), g2(5), g2(6), g3(4), g3(5), g3(6)]获取map的结果列表?

我考虑过使用以下代码:

map(lambda x:x(y), listFunc, ListArg)

但它只给出[g1(4), g2(5), g3(6)]的结果。

谢谢,

4 个答案:

答案 0 :(得分:5)

如果您想使用Option<R>,可以执行以下操作:

map()

答案 1 :(得分:5)

对于包含两个for-clauses的列表理解,这是一个完美的用例:

>>> def g1(x): return 1*x
... 
>>> def g2(x): return 2*x
... 
>>> def g3(x): return 3*x
... 
>>> funcs = [g1,g2,g3]
>>> args = [4,5,6]
>>> [f(a) for f in funcs for a in args]
[4, 5, 6, 8, 10, 12, 12, 15, 18]
>>> 

这非常具有可读性和显着功能 - 列表推导是从Haskell借来的。

如果您觉得有必要使用地图,那么您必须像这样接近它:

>>> import itertools
>>> list(map(lambda f,a : f(a), *zip(*itertools.product(funcs,args))))
[4, 5, 6, 8, 10, 12, 12, 15, 18]

显然不可读并且可能更慢。列表理解在这里获胜。

答案 2 :(得分:3)

您无法使用 map执行此操作,但您可以将一个调用嵌套在另一个调用中,并使用sum来平展结果。

>>> sum(map(lambda x : map(lambda f: f(x), ListArg),listFunc), [])
[5, 6, 7, 6, 7, 8, 7, 8, 9]

答案 3 :(得分:3)

itertools.product帮助

def g(y): 
    def f(x):
        return x+y
    return f
funcs = map(g, [1, 2, 3])
args = [4,5,6]
p = itertools.product(funcs, args)
r = [f(arg) for f, arg in p]

我能想出的列表理解的唯一替代方法是:

def foo(args):
    f, a = args
    return f(a)
r = list(map(foo, p)) # using p from above