在pandas
中使用rolling
函数时,是否可以包含开始和结束索引值,以使序列的值是周期性的?
例如,我有以下Series
实例:
import pandas as pd
timestamps = [1, 2, 3, 4]
quantities = [1, 16, 9, 4]
series = pd.Series(quantities, index=timestamps)
当我输入时:
series.rolling(window=3, min_periods=3, center=True).median()
收益 率:
1 NaN
2 9.0
3 9.0
4 NaN
dtype: float64
但是,我想在计算中包含序列的任一端,以便结果为:
1 median([4, 1, 16]) 4.0
2 median([1, 16, 9]) 9.0
3 median([16, 9, 4]) 9.0
4 median([9, 4, 1]) 4.0
dtype: float64
感谢您尝试实现这一目标的任何帮助!
我们可以使用第一个和最后一个元素扩展列表,然后计算滚动中位数。之后,我们dropna
并重新分配系列
不会赢得最漂亮代码的价格,但它可以完成工作。想不出其他办法。
quantities2 = quantities.copy()
quantities2.insert(0, quantities2[-1])
quantities2.insert(len(quantities2), quantities2[0])
print(quantities2)
[4, 1, 16, 9, 4, 4]
series = pd.Series(quantities2)
series = pd.Series(series.rolling(window=3, min_periods=3, center=True).median().dropna(), index=timestamps)
print(series)
1 4.0
2 9.0
3 9.0
4 4.0
dtype: float64
但话又说回来,我们必须写出最干净的代码吗?:)