如何创建或更新列

Question

我有一个数据框x:

item   S  sale
A1     1   5
A1     2   10
A2     1   3
A2     2   7
A2     3   8

我希望结果完全符合这种格式

item   S  sale  cumsum
A1     1   5      5
A1     2   10    15 
A2     1   3      3
A2     2   7      10
A2     3   8      18

我创建了一个字典，其中包含项和作为键值并使用 for 循环

x_dict=df.groupby(['item']).apply(lambda g: g.S.values).to_dict()

for i , j in x_dict.iteritems():

但我无法访问其给出错误的值：

（意外的 EOF 文件）

Answer 1

你可以试试groupby和cumsum：

df['cumsum'] =  df.groupby('item')['sale'].cumsum()
print df
  item  S  sale  cumsum
0   A1  1     5       5
1   A1  2    10      15
2   A2  1     3       3
3   A2  2     7      10
4   A2  3     8      18