Cách áp dụng danh sách tổng hợp (aggregation list) trên mỗi nhóm DataFrame pandas

Áp dụng danh sách tổng hợp dùng agg().

Nhập thư viện:

import pandas as pd

Tạo DataFrame:

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

Chỉ định list cho agg() sau groupby:

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

Ví dụ

import pandas as pd 
dataFrame = pd.DataFrame({"Car": ['BMW', 'Lexus', 'Lexus', 'Mustang', 'Bentley', 'Mustang'],
    "Units": [100, 150, 110, 80, 110, 90]}) 
print("DataFrame ...\n", dataFrame) 
dataFrame = dataFrame.groupby('Car').agg(list) 
print("\nDataFrame ...\n", dataFrame)

Kết quả

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]