Python-计算加权滚动标准偏差

Question

我有一分钟一分钟的熊猫数据帧df。我希望将加权应用于返回，并计算滚动加权均方差，窗口=10。我可以计算非加权性病，年化：

df_spy['10mVol'] = df_spy['Return'].rolling(center=False,window=10).std()*(1440*252)**(0.5)*100

在Numpy中还有另一个关于加权std的问题，但我对滚动加权std很好奇。（加权标准偏差，单位为NumPy？）

计算加权标准差的公式为：https://math.stackexchange.com/questions/320441/standard-deviation-of-the-weighted-mean

weighting   Midpoint   Return      10mVol   Weighted
0.2         215.6700    NaN         NaN      NaN
0.8         215.8400    -0.000788   NaN     -0.000630
0.8         216.0600    -0.001019   NaN     -0.000815

谢谢你的帮助

Answer 1

据我所知，rolling方法之后的链式函数是一个接受数组并给出数字的函数。为每个窗口计算该函数。因此，如果我们有一个计算加权标准差的函数，我们可以将它与lambda函数一起使用，得到滚动加权标准差。（我希望您提供的加权std计算没有出错）

import pandas as pd
import numpy as np


def weighted_std(values, weights):
    # For simplicity, assume len(values) == len(weights)
    # assume all weights > 0
    sum_of_weights = np.sum(weights)
    weighted_average = np.sum(values * weights) / sum_of_weights
    n = len(weights)
    numerator = np.sum(n * weights * (values - weighted_average) ** 2.0)
    denominator = (n - 1) * sum_of_weights
    weighted_std = np.sqrt(numerator / denominator)
    return weighted_std


def rolling_std(s, weights):
    window_size = len(weights)
    return s.rolling(center=False, window=window_size).apply(lambda win: weighted_std(win, weights))

s = pd.Series(np.random.random([10]))  # generate random data
w = np.array([1., 3., 5.])  # choose weights
print(s.values)
print(rolling_std(s, w).values)

示例输出：

[ 0.08101966  0.57133241  0.29491028  0.25139964  0.26151065  0.45768199
  0.94459935  0.21534497  0.35999294  0.60242746]
[        nan         nan  0.19701963  0.11936639  0.01539041  0.12097725
  0.33346742  0.40784167  0.25884732  0.17709334]

这里的lambda win:weighted_std（win，weights）是一个函数，它将数组作为输入并返回一个数字。