Python - Cách đặt lại chỉ mục (reset index) sau Groupby trong pandas

Để đặt lại chỉ mục sau groupby, trước tiên nhóm theo một cột sử dụng groupby(). Sau đó, sử dụng reset_index().

Trước tiên, nhập thư viện cần thiết:

import pandas as pd

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

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

Nhóm theo cột Car:

resDF = dataFrame.groupby("Car").mean()

Bây giờ, đặt lại chỉ mục sau khi nhóm:

resDF.reset_index()

Ví dụ

Dưới đây là mã:

import pandas as pd
# Tạo dataframe với hai cột
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)
# Nhóm theo Car
resDF = dataFrame.groupby("Car").mean()
print("\nDataFrame...\n", resDF)
# Đặt lại chỉ mục sau khi nhóm
print("\nReset index after grouping...\n", resDF.reset_index())

Kết quả

Điều này sẽ tạo ra đầu ra 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
DataFrame...
            Reg_Price
Car
Audi     1266.666667
Lexus    1450.000000
Mercedes 1183.333333
Reset index after grouping...
     Car   Reg_Price
0   Audi  1266.666667
1  Lexus  1450.000000
2 Mercedes 1183.333333