如何在 itertools.accumulate 中定义自己的 func 参数？

Question

itertools.accumulate的用法如下：

itertools.accumulate(iterable[, func, *, initial=None])

例如

list(itertools.accumulate([0, 7, 19, 13],operator.sub))
# [0, -7, -26, -39]

我使用减法运算符作为func 参数。

如何定义我自己的func 参数？

我希望它类似于operator.sub，但不是从第一项中减去第二项，而是从第二项中减去第一项。

输出[0, -7, -26, -39] 产生如下：

[0, (0) - (7), (0 - 7) - (19), ((0 - 7) - (19)) - (13)] == [0, -7, -26, -39]

我想要它：

[0, (7) - (0), (19) - (7 - 0), (13) - ((19) - (7 - 0))] == [0, 7, 12, 1]`

与operator.sub 相同，只是减法中的参数顺序被交换了。或者以其他方式说 - 如果 operator.sub 正在执行 x1-x2 我希望它在每个迭代步骤中执行 x2-x1。

Answer 1

创建一个有两个参数的函数，返回第二个减去第一个的结果：

from itertools import accumulate

def my_sub(a, b):
  return b - a

print(list(accumulate([0, 7, 19, 13], my_sub)))

输出：

[0, 7, 12, 1]

或者你可以使用lambda:

list(accumulate([0, 7, 19, 13], lambda a, b : b - a))

Answer 2

一切都是python中的对象，可以传递，包括函数。 func 可以是任何可调用对象，包括 lambda、函数、方法以及类具有 __call__ 方法的任意对象。

以下是这些选项的一些示例：

1. lambda 是一个特殊的单表达式函数，非常容易定义。它很方便，因为您可以匿名内联使用它：

    accumulate([0, 7, 19, 13], lambda a, b: b - a)
   

   由于一切都是对象，因此 lambda 不必是匿名的：

    func = lambda a, b: b - a
    accumulate([0, 7, 19, 13], func)
   

2. 函数可能是最普遍的可调用对象。您使用def 语句创建一个函数：

    def func(a, b):
        return b - a
    accumulate([0, 7, 19, 13], func)
   

   请注意，当您将其分配给名称时，它看起来很像 lambda。返回单个表达式的 Lambda 和函数在大多数方面的行为几乎完全相同。

   您也可以使用内置函数：

    accumulate([0, 7, 19, 13], operator.rsub)
   

   甚至大多数类方法都作为函数存在于类对象中。如果你有一个统一的列表，例如所有ints，你可以利用这个：

    accumulate([0, 7, 19, 13], int.__rsub__)
   

   这是因为int.__rsub__ 是一个接受两个参数的函数，它位于int 类中。当您执行(3).__rsub__(4) 时，您将该函数绑定到一个实例，这会将其转换为只有一个显式参数的方法。

3. 更深奥的可调用类型是具有__call__ 方法的类的实例：

    class Subtractor:
        def __call__(self, a, b):
            return b - a
    accumulate([0, 7, 19, 13], Subtractor())
   

   注意Subtractor() 中的括号。他们创建了一个可调用的新实例，因为该类定义了__call__。

4. 类函数对象的一个​​特例是staticmethod 和classmethod 对象。它们不是严格意义上的函数，而是将函数对象包装在一个类中。如果你有一个实用程序类，@staticmethod 装饰器可以应用于不依赖类或实例状态的函数：

    class Subtractor:
        @staticmethod
        def func1(a, b):
            return b - a
        @classmethod
        def func2(cls, a, b):
            return b - a
   
    accumulate([0, 7, 19, 13], Subtractor.func1)
    accumulate([0, 7, 19, 13], Subtractor().func2)
   

   func1 的行为类似于普通函数。事实上，它根本不会绑定到实例：您也可以使用accumulate([0, 7, 19, 13], Subtractor().func1)。但是，func2 必须绑定，但必须绑定到类而不是实例。您也可以拨打accumulate([0, 7, 19, 13], Subtractor.func2)。