时间戳

首页 > 解决方案 > 如何生成条件滚动平均值？

python - 如何生成条件滚动平均值？

问题描述

我希望能够根据在单独列中标记为 True 的值对 Data 列执行滚动平均值。下面是一个示例数据集。

                      Data        Condition
2020-01-01 0:00       2.0          False     
2020-01-01 0:01       2.0          False       
2020-01-01 0:02       732.1        True    
2020-01-01 0:03       734.2        True       
2020-01-01 0:04       733.3        True       
2020-01-01 0:05       731.2        True  
2020-01-01 0:06       2.0          False  
2020-01-01 0:07       2.0          False  
...
2020-06-15 23:52      2.0          False       
2020-06-15 23:53      732.1        True    
2020-06-15 23:54      734.2        True       
2020-06-15 23:55      733.3        True       
2020-06-15 23:56      731.2        True 
2020-06-15 23:57      2.0          False      
2020-06-15 23:58      2.0          False       
2020-06-15 23:59      2.0          False

对于数据列，我的预期输出（窗口大小为 3）如下所示：

[2.0, 2.0, 733.15, 733.2, 732.9, 732.25, 2, 2, ..., 2.0, 733.15, 733.2, 732.9, 732.25, 2.0, 2.0, 2.0]。

所以本质上，对于每一行，它会取该行上方和下方的行的平均值，以及有问题的行，但总是排除错误时的数字。

这是我找到的一些代码，除了它不适用于当前的熊猫。

pandas.rolling_mean(df[df.Condition], window=30) 我在这里找到它：滚动条件均值

任何有关这方面的帮助或信息将不胜感激！

标签： pythonpandasrolling-computation

解决方案

我相信您会过滤数据框以仅获取符合您条件的行，然后应用滚动平均值。

df['rolling'] = df.loc[df['Condition'] == True]['Data'].rolling(30).mean().fillna(0)

推荐阅读