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

Python Pandas - 원하는 텍스트가 포함된 행을 반복하고 가져옵니다.

<시간/>

원하는 텍스트가 포함된 행을 반복하고 가져오려면 itertuples() 및 find() 메서드를 사용하십시오. itertuples()는 DataFrame 행을 반복합니다.

먼저 별칭을 사용하여 필요한 라이브러리를 가져오도록 합시다 -

import pandas as pd

CSV는 아래 경로와 같이 바탕 화면에 있습니다 -

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

CSV 파일을 읽고 Pandas DataFrame을 생성합시다 -

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

특정 텍스트가 포함된 행을 반복하고 가져옵니다. "Lamborghini"라는 텍스트가 포함된 Car 열을 가져오고 있습니다. -

for k in dataFrame.itertuples():
   if k[1].find('Lamborghini') != -1:
      print(k)

예시

다음은 코드입니다.

import pandas as pd

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

# iterate and fetch the rows containing a specific text
# we are finding for Car column with text Lamborghini
for k in dataFrame.itertuples():
   if k[1].find('Lamborghini') != -1:
      print(k)

출력

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

Pandas(Index=5, Car='Lamborghini', Place='Chandigarh', UnitsSold=80)
Pandas(Index=8, Car='Lamborghini', Place='Delhi', UnitsSold=100)