无法使用这些索引器对 RangeIndex 进行位置索引 - numpy 单行获取系列中每 12 个值的平均值

Question

我正在尝试创建一个单行代码，用每 12 个先前值的平均值填充 pandas 系列中的 na 值。 我觉得这应该有效

test = pd.Series([10,5,3,4,6,12,6,7,2,4,7,12,6,7,2,4,7,12,3,4,6,12,6,7]+[pd.NA]*12)
test = pd.Series(np.where(test.isna(),test.iloc[test.index%12::12].mean(),test))

因为当我有 np.where() function 的第二部分使用“test.index%12”时，它返回我想用 iloc 对系列进行切片的数字，但我运行它时出错，因为测试.index 似乎正在拉动代表“测试”索引的整个系列 - 这是我得到的错误：

TypeError: cannot do positional indexing on RangeIndex with these indexers [Int64Index([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11,  0,  1,  2,  3,  4,
             5,  6,  7,  8,  9, 10, 11,  0,  1,  2,  3,  4,  5,  6,  7,  8,  9,
            10, 11],
           dtype='int64')] of type Int64Index

我怎样才能让 iloc 按预期对系列进行切片？ 或者如果有更好的方法来完成我想做的事情，请告诉我

Answer 1

如果你真的想要一个单线解决方案。 这是使用最后 12 个数字的平均值，不确定这是“前 12 个值的平均值”的意思

import pandas as pd
import numpy as np

test = pd.Series([10,5,3,4,6,12,6,7,2,4,7,12,6,7,2,4,7,12,3,4,6,12,6,7]+[np.nan]*12)

test.where(lambda x: ~np.isnan(x), test.rolling(12, min_periods=1).mean())

请注意，我已将pd.NA更改为np.nan