如何对每个 pandas DataFrame 组应用聚合列表？

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

import pandas as pd

创建一个有两列的 DataFrame −

dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Lexus', 'Mustang', 'Bentley', 'Mustang'],"Units": [100, 150, 110, 80, 110, 90]
   }
)

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

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

示例

以下是完整的代码 −

import pandas as pd

# Create DataFrame
dataFrame = pd.DataFrame(
   {
      "Car": ['BMW', 'Lexus', 'Lexus', 'Mustang', 'Bentley', 'Mustang'],"Units": [100, 150, 110, 80, 110, 90]
   }
)

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

# specifying list as argument using agg()
dataFrame = dataFrame.groupby('Car').agg(list)

# displaying dataframe lists
print("\nDataFrame ...\n",dataFrame)

输出

将生成以下输出 −

DataFrame ...
       Car   Units
0      BMW     100
1    Lexus     150
2    Lexus     110
3  Mustang      80
4  Bentley     110
5  Mustang      90

DataFrame ...
            Units
Car
BMW         [100]
Bentley     [110]
Lexus   [150,110]
Mustang   [80,90]

AmitDiwan

更新时间： 30-9-2021

102 次浏览

开启您的 职业

完成课程获得认证

开始