如何對每個 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]

更新時間: 30-9-2021

102 次瀏覽

開啟您的 職業

完成課程獲得認證

開始
廣告
© . All rights reserved.