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

更新於: 2021-09-28

428次瀏覽

開啟您的職業生涯

完成課程獲得認證

開始吧
廣告