在 Pandas Dataframe 中添加后續值

Question

我有以下 dataframe：

Week Total
1     5
2     6
3     10

我想讓第 2 周等於第 1 周 + 第 2 周，第 3 周等於第 1 周 + 第 2 周 + 第 3 周。我的 dataframe 比這更大，所以簡單地做添加不起作用，因為周數可以動態變化。 老實說，我不知道如何處理這個問題。 我在想也許做一個 for 循環，每周迭代並從當前減去最小的周數，以知道它需要添加多少周。 但這聽起來不像是一種干凈的方法。 有更好的解決方案嗎？

Answer 1

按周對行進行排序后使用cumsum （如果需要）：

out = (df
  .sort_values('Week')
  .assign(Total=lambda d: d['Total'].cumsum())
)

如果周已排序：

df['Total'] = df['Total'].cumsum()

output：

   Week  Total
0     1      5
1     2     11
2     3     21