python - 在 Pandas 中减少(折叠)

Question

Pandas DataFrame 是否有任何缩减/折叠实现？例如，我想获取数据帧 df 中名为 cost 的列中的数字总和，使用类似 lambda acc, x 的方法，其中 x 是一个 DataFrame 行。

我该怎么办？

附言我知道 .sum()，但还有许多其他可能的 \acc,x -> ... 函数。

Answer 1

可以通过以下方式实现快速折叠(将 plus 替换为您自己的函数):

import numpy as np
def accum(op,ser):
    u_op = np.frompyfunc(op, 2, 1) # two inputs, one output
    return u_op.accumulate(ser, dtype=np.object)
def plus(x,y):
    return x+y
accum(plus,np.arange(10))

你得到:

array([0, 1, 3, 6, 10, 15, 21, 28, 36, 45], dtype=object)

这适用于 numpy 系列，因此也适用于 pandas 数据帧的组件。如果有一个直接在数据帧上工作的解决方案会很有趣，这样就可以组合多个系列。