Pandas를 사용하여 특정 텍스트가 포함된 행 선택

특정 텍스트가 포함된 행을 선택하려면 contains() 메서드를 사용합니다. 다음이 CSV 파일 경로라고 가정해 보겠습니다. -

C:\\Users\\amit_\\Desktop\\SalesRecords.csv

먼저 CSV 파일을 읽고 Pandas DataFrame을 생성해 보겠습니다 -

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

이제 특정 텍스트 "BMW"가 포함된 행을 선택하겠습니다. -

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

예시

다음은 코드입니다 -

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 "BMW"
dataFrame = dataFrame[dataFrame['Car'].str.contains('BMW')]
print("\nFetching rows with text BMW ...\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 BMW ...
   Car   Place UnitsSold
3  BMW   Delhi 95