python - Pandas 通过聚合减法分组

Question

我有一个 pandas 数据框 df，其中包含帐户条目，例如人名、帐户 ID 具有贷方和借方条目，例如

date        Name      transaction-type  tran
2013-03-05  john Doe   credit          10
2013-05-05  john Doe   debit           20
2012-06-01  jane Doe   credit          50

我想按日期、名称和交易类型对交易进行分组并聚合交易？我怎么能这样做？我希望能够在 tran 列上执行 reduce(numpy.subtract)，但我不确定 Pandas 的正确语法。

Answer 1

IIUC，您只需要.groupby然后.sum()：

>>> df
                 date      Name transaction-type  tran
0 2013-03-05 00:00:00  john Doe           credit    10
1 2013-05-05 00:00:00  john Doe            debit    20
2 2012-06-01 00:00:00  jane Doe           credit    50
3 2012-06-01 00:00:00  jane Doe           credit    22
4 2012-06-02 00:00:00  jane Doe           credit    75
>>> df.groupby(["date", "Name", "transaction-type"]).sum()
                                      tran
date       Name     transaction-type      
2012-06-01 jane Doe credit              72
2012-06-02 jane Doe credit              75
2013-03-05 john Doe credit              10
2013-05-05 john Doe debit               20

请参阅文档中关于groupby 聚合的部分。

如果你想要总签名值，你也可以得到：

>>> df["tran"][df["transaction-type"] == "debit"] *= -1
>>> df.groupby(["date", "Name"]).sum()
                     tran
date       Name          
2012-06-01 jane Doe    72
2012-06-02 jane Doe    75
2013-03-05 john Doe    10
2013-05-05 john Doe   -20