Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

จะใช้รายการการรวมกับ 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]