Python - วิธีจัดกลุ่มแถว DataFrame ในรายการใน Pandas

หากต้องการจัดกลุ่มแถว dataframe เป็นรายการ ให้ใช้ฟังก์ชัน apply() ขั้นแรก ให้เรานำเข้าไลบรารีที่ร้องขอ -

import pandas as pd

สร้าง DataFrame ที่มี 2 คอลัมน์ -

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

การจัดกลุ่ม DataFrame ลงในรายการโดยใช้ apply(list) -

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

ตัวอย่าง

ต่อไปนี้เป็นรหัส -

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

ผลลัพธ์

สิ่งนี้จะสร้างผลลัพธ์ต่อไปนี้ -

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