Pandas：按条件分组

Question

我必须对不包含任何SmallSeller产品 (SellCategory)的交易 (InvoiceNo) 进行分组，但我不确定如何进行。

我最终必须比较Quantity*UnitPrice两种类型的交易（包含SmallSeller产品的交易与不包含任何SmallSeller产品的交易）产生的收入 ( )。

（这是我的数据示例）

    InvoiceNo   SellCategory    Revenue

      23        Medium-Seller     54

      23         Best-Seller     108

      23         Best-Seller      71

      24        Medium-Seller     43

      24         Small-Seller     18

      25         Best-Seller      89

      25         Small-Seller     23

      26        Medium-Seller     65

      26         Best-Seller      98

      26         Small-Seller     23

Answer 1

首先，我们可以选择相关行，然后Revenue使用 groupby on进行聚合InvoiceNo。最后，我们在最后连接两种数据帧并重命名列：df1 = pd.concat([df[df['SellCategory'] == 'Small-Seller'].groupby('InvoiceNo').sum(), df[df['SellCategory'] == 'Small-Seller'].groupby('InvoiceNo').sum()], 1)df1.columns = ['Revenue_Small_Seller', 'Revenue_without_Small_Seller']

Answer 2

您可以通过以下方式进行分组，然后进行进一步的操作：df[df.SellCategory != 'Small-Seller'].groupby('InvoiceNo')

Answer 3

df_pivot=df.pivot_table(index='InvoiceNo',columns='SellCategory',values='Revenue',aggfunc='sum').reset_index()