home

หน้าแรก
 Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

วิธีใช้ Aggregation List บนทุก Group ของ Pandas DataFrame

บทนำ

ใช้ agg(list) ร่วมกับ groupby() รวมค่าในแต่ละกลุ่มเป็นลิสต์

ขั้นตอน

  1. นำเข้า pandas
  2. สร้าง DataFrame (Car, Units) มี Car ซ้ำ
  3. จัดกลุ่มและรวมเป็นลิสต์: 
    dataFrame = dataFrame.groupby('Car').agg(list)

โค้ดเต็ม

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)

ผลลัพธ์

ได้ DataFrame Index เป็น Car คอลัมน์ Units เป็นลิสต์: BMW [100], Bentley [110], Lexus [150,110], Mustang [80,90]