不带零的滚动平均值和标准偏差

Question

我有一个数据框，其中一列表示在这个时间戳中生产了多少玉米。 比如说

timestamp corns_produced another_column
1           5                  4
2           0                  1
3           0                  3
4           3                  4
 

数据帧很大。。100000多行

我想计算moving average和std的1000个时间戳corn_produced。 幸运的是，使用rolling非常简单：

• my_df.rolling(1000).mean()
• my_df.rolling(1000).std()

但问题是我想忽略零，这意味着如果在最后1000个时间戳中只有5个玉米生产实例，我想对这5个元素进行mean和std

如何忽略零

我只是想澄清一下，我不想做下面的x = my_df[my_df['corns_produced'] != 0]，而不是在x上做rolling，因为它忽略了时间戳，不会给出我需要的结果

Answer 1

您可以使用^{}：

print (my_df.rolling(1000).apply(lambda x: x[x!= 0].mean()))
print (my_df.rolling(1000).apply(lambda x: x[x!= 0].std()))