Computer >> 컴퓨터 >  >> 프로그램 작성 >> Python

행 필터링 – Python Pandas

<시간/>

행을 필터링하고 특정 열 값을 가져오려면 Pandas contains() 메서드를 사용합니다. 먼저 alias −

를 사용하여 필요한 라이브러리를 가져옵니다.
import pandas as pd

read_csv()를 사용하여 CSV 파일을 읽습니다. 우리의 CSV 파일은 바탕 화면에 있습니다 −

dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\CarRecords.csv")

이제 특정 텍스트로 행을 필터링해 보겠습니다 -

dataFrame = dataFrame[dataFrame['Car'].str.contains('Lamborghini')]

예시

다음은 코드입니다.

import pandas as pd

# reading csv file
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\CarRecords.csv")
print("DataFrame...\n",dataFrame)

# select rows containing text "Lamborghini"
dataFrame = dataFrame[dataFrame['Car'].str.contains('Lamborghini')]
print("\nFetching rows with text Lamborghini ...\n",dataFrame)

출력

이것은 다음과 같은 출력을 생성합니다 -

DataFrame...
           Car       Place   UnitsSold
0         Audi   Bangalore          80
1      Porsche      Mumbai         110
2   RollsRoyce        Pune         100
3          BMW       Delhi          95
4     Mercedes   Hyderabad          80
5  Lamborghini  Chandigarh          80
6         Audi      Mumbai         100
7     Mercedes        Pune         120
8  Lamborghini       Delhi         100

Fetching rows with text Lamborghini ...
           Car       Place   UnitsSold
5  Lamborghini  Chandigarh          80
8  Lamborghini       Delhi         100