优雅地应用Python函数列表

时间:2016-08-24 12:29:57

标签: python functional-programming

我有一个输入值val和一个要按顺序应用的函数列表:

funcs = [f1, f2, f3, ..., fn]

如何优雅地申请而不是写作

fn( ... (f3(f2(f1(val))) ... )

并且也没有使用for循环:

tmp = val
for f in funcs:
    tmp = f(tmp)

感谢Martijn的精彩回答。我找到了一些阅读材料:https://mathieularose.com/function-composition-in-python/

2 个答案:

答案 0 :(得分:18)

使用reduce() function

# forward-compatible import
from functools import reduce

result = reduce(lambda res, f: f(res), funcs, val)

reduce()将第一个参数(一个可调用的)应用于从第二个参数中取出的每个元素,加上到目前为止的累积结果(如(result, element))。第三个参数是一个起始值(否则将使用funcs中的第一个元素)。

在Python 3中,内置函数已移至functools.reduce() location;为了向前兼容,Python 2.6及更高版本中提供了相同的参考。

其他语言可能会调用此folding

如果您还需要每个函数的中间结果,请使用itertools.accumulate()(仅从Python 3.3开始,以获取带有函数参数的版本):

from itertools import accumulate, chain
running_results = accumulate(chain(val, funcs), lambda res, f: f(res))

答案 1 :(得分:3)

MartijnPieters的回答非常好。我要添加的唯一内容是这称为function composition

为这些泛型赋予名称意味着您可以在需要时使用它们

from functools import reduce

def id(x):
  return x

def comp(f,g):
  return lambda x: f(g(x))

def compose(*fs):
  return reduce(comp, fs, id)

# usage
# compose(f1, f2, f3, ..., fn) (val)

print(compose (lambda x: x + 1, lambda x: x * 3, lambda x: x - 1) (10))
# = ((10 - 1) * 3) + 1
# = 28

有时你希望它看起来更好 -

def seq (x): return lambda k: seq (k (x))

def fn1 (x): return x - 1
def fn2 (x): return x * 3
def fn3 (x): return x + 1

seq (10) (fn1) (fn2) (fn3) (print) # 28
         # 10    9    27    28

有时你想要更多的灵活性 -

from operator import add, mul, sub

def seq (x):
  return lambda k, *v: seq (k (x, *v))

seq (10) (sub, 1) (mul, 3) (add, 1) (print) # 28
          # 10       9        27       28

seq (1) (add, 2) (add, 3) (add, 4) (add, 5) (print) # 15
        # 1        3        6        10       15

sum = seq(0)
for x in range(10):
  sum = sum (add, x)
sum (print) # 45