Computer >> Máy Tính >  >> Lập trình >> Python

Python - Nhóm các cột trong Khung dữ liệu Pandas

Để nhóm các cột trong khung dữ liệu Pandas, hãy sử dụng groupby (). Đầu tiên, hãy để chúng tôi tạo khung dữ liệu Pandas -

dataFrame = pd.DataFrame(
   {
      "Car": ["Audi", "Lexus", "Audi", "Mercedes", "Audi", "Lexus", "Mercedes", "Lexus", "Mercedes"],
      "Reg_Price": [1000, 1400, 1100, 900, 1700, 1800, 1300, 1150, 1350]
   }
)

Bây giờ chúng ta hãy phân nhóm theo chuyên mục Xe -

res = dataFrame.groupby("Car")

Sau khi phân nhóm, chúng ta sẽ sử dụng các hàm để tìm phương tiện Giá đăng ký (Reg_Price) của các tên xe được nhóm -

res.mean()

Điều này tính toán trung bình của Giá đăng ký theo cột Xe ô tô.

Ví dụ

Sau đây là mã -

import pandas as pd

# dataframe with one of the columns as Reg_Price
dataFrame = pd.DataFrame(
   {
      "Car": ["Audi", "Lexus", "Audi", "Mercedes", "Audi", "Lexus", "Mercedes", "Lexus", "Mercedes"],
      "Reg_Price": [1000, 1400, 1100, 900, 1700, 1800, 1300, 1150, 1350]
   }
)

print"DataFrame...\n",dataFrame

# grouped according to Car
res = dataFrame.groupby("Car")

print"\nMean of Registration Price grouped according to Car names...\n",res.mean()

Đầu ra

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

DataFrame...
      Car    Reg_Price
0     Audi        1000
1    Lexus        1400
2     Audi        1100
3 Mercedes         900
4     Audi        1700
5    Lexus        1800
6 Mercedes        1300
7    Lexus        1150
8 Mercedes        1350
Mean of Registration Price grouped according to Car names...
           Reg_Price
    Car
    Audi   1266.666667
   Lexus   1450.000000
Mercedes   1183.333333