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

Python - Làm thế nào để đặt lại chỉ mục sau Groupby gấu trúc?

Để đặt lại chỉ mục sau khi nhóm theo, nhóm đầu tiên theo một cột bằng cách sử dụng groupby (). Sau đó, sử dụng reset_index ().

Đầu tiên, hãy nhập thư viện được yêu cầu -

import pandas as pd

Tạo DataFrame có 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 chuyên mục Xe -

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

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

resDF.reset_index()

Ví dụ

Sau đây là mã -

import pandas as pd

# creating a dataframe with two columns
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
resDF = dataFrame.groupby("Car").mean()

print"\nDataFrame...\n", resDF

# resetting index after grouping
print"\nReset index after grouping...\n", resDF.reset_index()

Đầ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

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