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

Python - วิธี Reset Index หลัง Groupby ใน Pandas


เพื่อ Reset Index หลัง Groupby เริ่มต้นด้วยการ Group ตามคอลัมน์โดยใช้ groupby() จากนั้นใช้ reset_index()

ก่อนอื่น import ไลบรารีที่จำเป็น -

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]
    }
)

Group ตามคอลัมน์ Car -

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

ตอนนี้ Reset Index หลัง Grouping -

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