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
廣告