Python - Làm cách nào để nhóm các hàng DataFrame thành danh sách trong Pandas?

Để nhóm các hàng khung dữ liệu vào danh sách, hãy sử dụng hàm apply (). Đầu tiên, hãy để chúng tôi nhập thư viện yêu cầu -

import pandas as pd

Tạo DataFrame với 2 cột -

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

Nhóm DataFrame thành danh sách với áp dụng (danh sách) -

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

Ví dụ

Sau đây là mã -

import pandas as pd

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

print"DataFrame ...\n",dataFrame

# grouping dataframe into lists
dataFrame = dataFrame.groupby('Car')['Units'].apply(list)

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

Đầu ra

Điều này sẽ tạo ra kết quả sau -

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