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

Python - Tính toán đầu tiên của các giá trị nhóm trong Pandas DataFrame

Để tính toán giá trị đầu tiên của nhóm, hãy sử dụng phương thức groupby.first (). Đầu tiên, hãy nhập thư viện được yêu cầu với bí danh -

import pandas as pd;

Tạo DataFrame có 3 cột -

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

Bây giờ, hãy nhóm DataFrame theo một cột -

groupDF = dataFrame.groupby("Car")

Tính toán giá trị đầu tiên của nhóm và đặt lại chỉ mục -

res = groupDF.first()
res = res.reset_index()

Ví dụ

Sau đây là mã hoàn chỉnh -

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)

# grouping DataFrame by column Car
groupDF = dataFrame.groupby("Car")

res = groupDF.first()
res = res.reset_index()

print("\nFirst of group values = \n",res)

Đầu ra

Điều này sẽ tạo ra kết quả sau. Lần xuất hiện đầu tiên của các giá trị lặp lại được hiển thị, tức là giá trị đầu tiên của nhóm -

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

First of group values =
     Car      Place   Units
0    BMW      Delhi     100
1  Lexus  Bangalore     150
2  Tesla     Punjab      80