python - NumPy 中的 cumsum 函数在添加时会衰减吗？

Question

我有一个值数组 a = (2,3,0,0,4,3)

y=0
for x in a:
  y = (y+x)*.95

有什么方法可以在 numpy 中使用 cumsum 并在添加下一个值之前对每一行应用 .95 衰减？

Answer 1

您要求的是一个简单的 IIR Filter . Scipy 的 lfilter()是为此而生的:

import numpy as np
from scipy.signal import lfilter

data = np.array([2, 3, 0, 0, 4, 3], dtype=float)  # lfilter wants floats

# Conventional approach:
result_conv = []
last_value = 0
for elmt in data:
    last_value = (last_value + elmt)*.95
    result_conv.append(last_value)

# IIR Filter:
result_IIR = lfilter([.95], [1, -.95], data)

if np.allclose(result_IIR, result_conv, 1e-12):
    print("Values are equal.")