python - 不使用掩码值的移动平均线

Question

我想知道您是否可以给我一种方法来计算 numpy 掩码数组上的滚动均值，而不使用掩码值。我目前正在使用 convolve numpy 方法，但它不起作用：

import numpy as np

a=np.array([1,2,5,4,9,6,1000,3,6,2,9,0])
a=np.ma.masked_where(a>990,a)
print(a)
a=np.convolve(a, np.ones(5), "valid")/5
print(a)

>>>[1 2 5 4 9 6 -- 3 6 2 9 0]
>>>[  4.2   5.2 204.8 204.4 204.8 203.4 204.    4. ]

非常感谢您的回答。

Answer 1

>>> a = np.array([1,2,5,4,9,6,1000,3,6,2,9,0])
>>> b = np.delete(a, a>990)
>>> window_size = 5
>>> np.convolve(b, np.ones(window_size)/window_size, "valid")
array([4.2, 5.2, 5.4, 5.6, 5.2, 5.2, 4. ])