Python Pandas-根據給定的窗口並從特定值開始計算特定列的總和

Question

我正在嘗試在Python Pandas數據框中創建一個新列，其中包含指定范圍內並從某一行開始的另一列的求和值。 例如，在下面的數據框中，我希望跟蹤1的值列的總和從第3行開始（使用索引行號）。但是，我還需要對跟蹤2重復相同的條件。此外，我需要從每個跟蹤的相同樣本編號開始。 有任何想法嗎？

Answer 1

IIUC，

groupby + rolling + sum

df = pd.DataFrame({'trace': [1]*5 + [2]*5, 'sample': list(range(1,6))*2,
                  'value': [0.25, 0.63, 0.98, 0.48, 0.52,0.79, 0.63, 0.11, 0.29, 0.81]})
df.groupby('trace').value.rolling(4).sum()

1      0     NaN
       1     NaN
       2     NaN
       3    2.34
       4    2.61
2      5     NaN
       6     NaN
       7     NaN
       8    1.82
       9    1.84