熊猫：在列中添加行

Question

我有一个这样的数据框

Count
1
0
1
1
1

我想在计数列中添加N和N + 1 并将其存储在N中，可以用熊猫的方式吗？

结果应该这样，从技术上讲，它是累计和：

Counts
1
1
2
3
4

Answer 1

您可以使用累积求和函数cumsum（）。

df = pd.DataFrame([1, 0, 1, 1,1], columns=['Count'])

df['Counts'] = df['Count'].cumsum()
print(df)

为您提供所需的输出。

   Count  Counts
0      1       1
1      0       1
2      1       2
3      1       3
4      1       4