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

Python - วิธีรีเซ็ตดัชนีหลังจาก Groupby pandas?


หากต้องการรีเซ็ตดัชนีตามกลุ่ม โดยที่กลุ่มแรกตามคอลัมน์โดยใช้ groupby() หลังจากนั้น ให้ใช้ reset_index()

ขั้นแรก นำเข้าไลบรารีที่จำเป็น -

import pandas as pd

สร้าง DataFrame ที่มี 2 คอลัมน์ -

dataFrame = pd.DataFrame(
   {
      "Car": ["Audi", "Lexus", "Audi", "Mercedes", "Audi", "Lexus", "Mercedes", "Lexus", "Mercedes"],

      "Reg_Price": [1000, 1400, 1100, 900, 1700, 1800, 1300, 1150, 1350]
   }
)

จัดกลุ่มตามคอลัมน์รถยนต์ −

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

ตอนนี้ รีเซ็ตดัชนีหลังจากจัดกลุ่ม -

resDF.reset_index()

ตัวอย่าง

ต่อไปนี้เป็นรหัส -

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()

ผลลัพธ์

สิ่งนี้จะสร้างผลลัพธ์ต่อไปนี้ -

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