python - 如何获得pandas数据帧中k个连续行的总和？

Question

我有这个 Pandas 数据框:

ts = pd.Series([1, 2, 3, 4, 5, 6, 7, 8])

我想得到的是一个数据框，其中包含另一列，其中四行的总和为 ts .总和应放在连续行的每一行中。
在这种情况下，这意味着一个新的数据框应该是这样的:

index ts sum
0 1 10
1 2 10
2 3 10
3 4 10
4 5 26
5 6 26
6 7 26
7 8 26

我怎么能这样做？

Answer 1

使用 GroupBy.transform 用整数除以 k :

k = 4
a = ts.groupby(ts.index // k).transform('sum')
#alternative if not RangeIndex
#a = ts.groupby(np.arange(len(ts)) // k).transform('sum')
print (a)
0    10
1    10
2    10
3    10
4    26
5    26
6    26
7    26
dtype: int64
    

如果需要 2 列的 DataFrame 添加 Series.to_frame :

df = ts.to_frame('ts')
df['sum'] = df.groupby(ts.index // k).transform('sum')
print (df)
   ts  sum
0   1   10
1   2   10
2   3   10
3   4   10
4   5   26
5   6   26
6   7   26
7   8   26