Pandas 对具有几乎相同值的行求和

Question

我正在尝试组织一个电子表格来跟踪项目流。目标是对具有相同操作、价格和日期的项目的金额列求和。例如：Item  action amount price datesocks   buy    10    $20  5/1socks   buy     5    $20  5/1socks   sell    5    $20  5/1shoes   sell    7    $25  5/2shoes   sell    2    $25  5/2shoes   sell    8    $30  5/2--会变成--socks   buy     15   $20  5/1socks   sell     5   $20  5/1shoes   sell     9   $25  5/2shoes   sell     8   $30  5/2这可能使用熊猫吗？

Answer 1

用于DataFrame.groupby对数据帧进行分组，'Item', 'action', 'price', 'date'然后使用 agg 函数sum计算amount每组列的总和，然后用于DataFrame.reset_index重置分组数据帧的索引：

df = df.groupby(['Item', 'action', 'price', 'date']).sum().reset_index().reindex(columns=df.columns)

结果：

# print(df)

   Item action  amount price date

0  shoes   sell       9   $25  5/2

1  shoes   sell       8   $30  5/2

2  socks    buy      15   $20  5/1

3  socks   sell       5   $20  5/1