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