python - 如何计算前缀行的平均值并将其作为熊猫中的新列？

Question

我有以下数据格式：

  col_a col_b
0   10   12
1   11   14
2   14   15
3   15   16
...

我想采用以下格式：（2个前缀数据作为每列的平均结果）

  col_a col_b  mean_col_a  mean_col_b
0   10   12       NAN          NAN
1   11   14       NAN          NAN
2   24   15       10.5         13
3   15   16       19.5         14.5
...

标签： pythonpandasnumpy

Answer 1

您可以join与rolling.

df.join(df.rolling(2).mean(), rsuffix='_mean')

   col_a  col_b  col_a_mean  col_b_mean
0     10     12         NaN         NaN
1     11     14        10.5        13.0
2     14     15        12.5        14.5
3     15     16        14.5        15.5