使用python的movvar matlab函数等价

Question

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

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

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