Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Python Pandas – Lấy hàng cuối cùng (last) của mỗi nhóm groupby

Giới thiệu

Để lấy hàng cuối cùng của mỗi nhóm sau khi groupby, dùng groupby.last() kết hợp reset_index() để chuyển index thành cột.

Ví dụ minh họa

import pandas as pd

dataFrame = pd.DataFrame(
    {
        "Car": ['BMW', 'Lexus', 'BMW', 'Tesla', 'Lexus', 'Tesla'],
        "Place": ['Delhi', 'Bangalore', 'Pune', 'Punjab', 'Chandigarh', 'Mumbai'],
        "Units": [100, 150, 50, 80, 110, 90]
    }
)
print("DataFrame ...\n", dataFrame)

# Groupby Car
groupDF = dataFrame.groupby("Car")
res = groupDF.last()
res = res.reset_index()
print("\nHàng cuối cùng của mỗi nhóm (last):\n", res)

Kết quả đầu ra

DataFrame ...
      Car       Place  Units
0     BMW      Delhi    100
1   Lexus  Bangalore    150
2     BMW      Pune     50
3   Tesla    Punjab     80
4   Lexus Chandigarh    110
5   Tesla    Mumbai     90

Hàng cuối cùng của mỗi nhóm (last):
     Car      Place  Units
0   BMW      Pune     50
1  Lexus  Chandigarh   110
2  Tesla    Mumbai     90

Giải thích: Nhóm BMW: hàng cuối là index 2 (Pune, 50). Lexus: index 4 (Chandigarh, 110). Tesla: index 5 (Mumbai, 90).