对一列执行累积求和，但如果求和在 Pandas 中变为负数，则重置为 0

Question

我有一个带有两列的 pandas 数据框，

Item    Value
0   A   7
1   A   2
2   A   -6
3   A   -70
4   A   8
5   A   0

我想在 Value 列上累积总和。但是，如果值变为负数，则在创建累积和时，我想将其重置回 0。

我目前正在使用如下所示的循环来执行此操作，

sum_ = 0
cumsum = []

for val in sample['Value'].values:
    sum_ += val
    if sum_ < 0:
        sum_ = 0
    cumsum.append(sum_)

print(cumsum) # [7, 9, 3, 0, 8, 8]

我正在寻找一种更有效的方法来在纯熊猫中执行此操作。

Answer 1

这只是 WeNYoBen 的评论。

如果您可以避免使用列表，通常建议您避免使用它。

示例

from numba import njit
import numpy as np

#with lists
@njit()
def cumli(x, lim):
    total = 0
    result = []
    for i, y in enumerate(x):
        total += y
        if total < lim:
            total = 0
        result.append(total)
    return result

#without lists
@njit()
def cumli_2(x, lim):
    total = 0.
    result = np.empty_like(x)
    for i, y in enumerate(x):
        total += y
        if total < lim:
            total = 0.
        result[i]=total
    return result

时间

没有 Numba（注释掉@njit()）：

x=(np.random.rand(1_000)-0.5)*5

  %timeit a=cumli(x, 0.)
  220 µs ± 2.25 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)
  %timeit a=cumli_2(x, 0.)
  227 µs ± 1.95 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

使用列表或数组没有区别。但如果你对这个函数进行 Jit 编译，情况就不是这样了。

使用 Numba：

  %timeit a=cumli(x, 0.)
  27.4 µs ± 210 ns per loop (mean ± std. dev. of 7 runs, 10000 loops each)
  %timeit a=cumli_2(x, 0.)
  2.96 µs ± 32.5 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)

即使在更复杂的情况下（最终数组大小未知，或者只知道最大数组大小），分配一个数组并在最后缩小它通常是有意义的，或者在简单的情况下，甚至运行一次算法即可知道最终的数组大小，而不是真正的计算。

Answer 2

稍微修改一下这个方法也比较慢，numba解决方案

sumlm = np.frompyfunc(lambda a,b: 0 if a+b < 0 else a+b,2,1)
newx=sumlm.accumulate(df.Value.values, dtype=np.object)
newx
Out[147]: array([7, 9, 3, 0, 8, 8], dtype=object)

---

numba解决方案

from numba import njit
@njit
def cumli(x, lim):
    total = 0
    result = []
    for i, y in enumerate(x):
        total += y
        if total < lim:
            total = 0
        result.append(total)
    return result
cumli(df.Value.values,0)
Out[166]: [7, 9, 3, 0, 8, 8]