使用 python 的 movvar matlab 函数的等价性

Question

使用python/numpy/pandas计算移动平均值的movvar(A, k, w) MATLAB函数等价于什么？

我尝试 pandas.rolling.mean() 使用 Matlab 文档中的这个示例，但结果不一致：

A= [4, 8, 6, -1, -2, -3, -1, 3, 4, 5]
a = pd.Series(A)

print(a.rolling(window=3).mean())

0         NaN
1    6.000000
2    4.333333
3    1.000000
4   -2.000000
5   -2.000000
6   -0.333333
7    2.000000
8    4.000000
9         NaN
dtype: float64

# using Matlab
M = movvar(A, k=3, w=1)

M = 1×10

4.0000    2.6667   14.8889   12.6667    0.6667    0.6667    6.2222    4.6667    0.6667    0.2500

Answer 1

您需要 rolling.var 并在 rolling 中指定 center=True 和 min_periods=1，在 var

中指定 ddof=0

print(a.rolling(3, center=True, min_periods=1).var(ddof=0))
0     4.000000
1     2.666667
2    14.888889
3    12.666667
4     0.666667
5     0.666667
6     6.222222
7     4.666667
8     0.666667
9     0.250000
dtype: float64

使用 python 的 movvar matlab 函数的等价性

Equivalence of movvar matlab function using python

python

matlab

numpy

pandas