python - 在numpy中淡出信号

Question

生成累积和的最惯用方法是什么，该累积和会随着移动而“淡出”。让我用一个例子来解释。

>>> np.array([1,0,-1,0,0]).cumsum()
array([1, 1, 0, 0, 0], dtype=int32)

但我想提供一个因素<1并产生类似的东西:

>>> np.array([1,0,-1,0,0]).cumsum_with_factor(0.5)
array([1.0, 0.5, -0.75, -0.375, -0.1875], dtype=float64)

速度快是一大优势!

Answer 1

你的结果可以通过线性卷积得到:

signal = np.array([1,0,-1,0,0])
kernel = 0.5**np.arange(5)
np.convolve(signal, kernel, mode='full')
# array([ 1.    ,  0.5   , -0.75  , -0.375 , -0.1875, -0.125 , -0.0625,
          0.    ,  0.    ])

如果性能是一个考虑因素，请使用 scipy.signal.fftconvolve，它是相同逻辑的更快实现。