Python - Gom nhóm hàng DataFrame thành danh sách (list) trong Pandas

Sử dụng groupby() kết hợp apply(list) để gom nhóm các hàng theo một cột và gom giá trị cột khác thành danh sách.

Import pandas:

import pandas as pd

Tạo DataFrame:

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

Gom nhóm theo 'Car' và áp dụng list cho cột 'Units':

dataFrame = dataFrame.groupby('Car')['Units'].apply(list)

Ví dụ hoàn chỉnh

import pandas as pd
dataFrame = pd.DataFrame({
    "Car": ['BMW', 'Lexus', 'Audi', 'Mustang', 'Bentley', 'Jaguar'],
    "Units": [100, 150, 110, 80, 110, 90]
})
print("DataFrame ...\n", dataFrame)
dataFrame = dataFrame.groupby('Car')['Units'].apply(list)
print("\nDataFrame sau gom nhóm thành list:\n", dataFrame)

Kết quả

DataFrame ...
     Car  Units
0   BMW    100
1  Lexus    150
2   Audi    110
3 Mustang     80
4 Bentley    110
5 Jaguar     90
DataFrame ...
Car
Audi      [110]
BMW       [100]
Bentley   [110]
Jaguar     [90]
Lexus     [150]
Mustang    [80]
Name: Units, dtype: object