Python Pandas——如何基於條件選擇DataFrame行
我們可以設定條件並提取DataFrame行。這些條件可以使用邏輯運算子甚至關係運算符來設定。
首先,匯入所需的pandas庫−
import pandas as pd
讓我們建立一個DataFrame並讀取我們的CSV檔案−
dataFrame = pd.read_csv("C:\Users\amit_\Desktop\SalesRecords.csv")
使用關係運算符提取註冊價格小於1000的資料幀行−
dataFrame[dataFrame.Reg_Price < 1000]
例項
程式碼如下−
import pandas as pd # reading csv file dataFrame = pd.read_csv("C:\Users\amit_\Desktop\SalesRecords.csv") print("DataFrame...\n",dataFrame) # count the rows and columns in a DataFrame print("\nNumber of rows and column in our DataFrame = ",dataFrame.shape) # fetching dataframe rows with registration price less than 1000 resData = dataFrame[dataFrame.Reg_Price < 1000] print("DataFrame...\n",resData)
輸出
這將生成以下輸出−
DataFrame... Car Date_of_Purchase Reg_Price 0 BMW 10/10/2020 1000 1 Lexus 10/12/2020 750 2 Audi 10/17/2020 750 3 Jaguar 10/16/2020 1500 4 Mustang 10/19/2020 1100 5 Lamborghini 10/22/2020 1000 Number of rows and column in our DataFrame = (6, 3) DataFrame... Car Date_of_Purchase Reg_Price 1 Lexus 10/12/2020 750 2 Audi 10/17/2020 750
廣告