Python — 計算 Pandas DataFrame 中組的第一項值


要計算組的第一項值,請使用 groupby.first() 方法。首先,使用別名匯入所需庫 −

import pandas as pd;

建立一個有 3 列的 DataFrame −

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

現在,按某列對 DataFrame 進行分組 −

groupDF = dataFrame.groupby("Car")

計算組的第一項值並重置索引 −

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

示例

以下是完整程式碼 −

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)

輸出

這將產生以下輸出。將顯示重複值的首次出現,即組的第一項值 −

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

更新於: 2021-09-29

120 次瀏覽

開啟您的 職業生涯

完成課程以獲得認證

開始學習
廣告