python panads组通过丢失列名

Question

共有3列，

ACCOUNT, DATE, AMOUNT

我正尝试在熊猫中执行以下操作，并计划在按输出数据帧分组之后使用“求和”列来做更多工作：

df_group_by=df.groupby('account').agg({'amount': [np.size, np.sum]},as_index=False).reset_index() # equal to "SELECT ACCOUNT, SUM(amount) as sum FROM table GROUP BY ACCOUNT;"

print (df_group['sum'])

然后出现以下错误消息：

KeyError: 'sum

我不明白。有人可以帮忙吗？谢谢！

我在python 3.6和pandas 0.19.2中

Answer 1

df_group是一个多索引数据框，您可以通过df_group ['amount'] ['sum']

访问'sum'列

Answer 2

此作品有效：

Pandas Groupby and Sum Only One Column

df_group_by=df.groupby(['account'], as_index=False)['amount'].sum()

print (df_group['amount'])