如何在Pandas DataFrame的每个分组上应用聚合列表？

要应用聚合列表，请使用agg()方法。首先，导入所需的库−

import pandas as pd

创建具有两列的DataFrame−

dataFrame = pd.DataFrame(
   {
      "Car": ['宝马', '雷克萨斯', '雷克萨斯', '野马', '宾利', '野马'],"Units": [100, 150, 110, 80, 110, 90]
   }
)

使用agg()指定列表作为参数−

dataFrame = dataFrame.groupby('Car').agg(list)

示例

以下是完整代码−

import pandas as pd

# 创建DataFrame
dataFrame = pd.DataFrame(
   {
      "Car": ['宝马', '雷克萨斯', '雷克萨斯', '野马', '宾利', '野马'],"Units": [100, 150, 110, 80, 110, 90]
   }
)

print("DataFrame ...\n",dataFrame)

# 使用agg()指定列表作为参数
dataFrame = dataFrame.groupby('Car').agg(list)

# 显示DataFrame列表
print("\nDataFrame ...\n",dataFrame)

输出

这将产生以下输出−

DataFrame ...
       Car   Units
0      宝马     100
1    雷克萨斯     150
2    雷克萨斯     110
3     野马       80
4     宾利       110
5     野马       90

DataFrame ...
            Units
Car
宝马         [100]
宾利         [110]
雷克萨斯   [150,110]
野马      [80,90]