python - 如何从 pandas 的累积总和中获取实际值

Question

假设，我有一个 DataFrame df。

>>> DATA = {'id':[1,2,3,4,5], 'salary':[1200,2300,2400,3620,2100] } 
>>> df = DataFrame(DATA)
>>> df
   id  salary
0   1    1200
1   2    2300
2   3    2400
3   4    3620
4   5    2100

从这个DataFrame df中，我可以得到一个新的DataFrame df1，其中包含工资的累计总和

>>> df['salary'] = df['salary'].cumsum() 
>>> df
   id  salary
0   1    1200
1   2    3500
2   3    5900
3   4    9520
4   5   11620

这是非常常见的情况。

现在，如果给我 df1 并且我必须找到 df，该怎么办？

   id  salary            id  salary
0   1    1200             0   1    1200
1   2    3500             1   2    2300
2   3    5900     ==>     2   3    2400
3   4    9520             3   4    3620
4   5   11620             4   5    2100

我必须从所有 id 的累计总和中找到其实际工资。

Answer 1

>>> df
   id  salary
0   1    1200
1   2    3500
2   3    5900
3   4    9520
4   5   11620
>>> df['salary'] = df['salary'].diff().fillna(df['salary'].iloc[0])
>>> df
   id  salary
0   1    1200
1   2    2300
2   3    2400
3   4    3620
4   5    2100

虽然 .fillna 在这里效率不高，因为它只是第一个出现 null 的值。因此，您只需将 .iloc[0] 处的第一个值替换为累积值即可。