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

주어진 DataFrame에서 'k'로 시작하는 도시 및 주 이름을 새 CSV 파일에 저장하는 프로그램을 Python으로 작성하십시오.

<시간/>

입력 -

City 및 State 열이 있는 DataFrame이 있고 아래와 같이 도시, 주 이름이 'k'로 시작하고 다른 CSV 파일에 저장한다고 가정합니다. -

City,State
Kochi,Kerala

해결책

이 문제를 해결하기 위해 다음 단계를 따릅니다.

  • DataFrame 정의

  • 아래 정의된 'k'로 시작하는 도시를 확인하십시오.

df[df['City'].str.startswith('K') & df['State'].str.startswith('K')]
  • 마지막으로 아래와 같이 'CSV' 파일에 데이터를 저장합니다.

df1.to_csv(‘test.csv’)

예시

더 나은 이해를 위해 다음 구현을 살펴보겠습니다.

import pandas as pd
import random as r
data = { 'City': ['Chennai','Kochi','Kolkata'],'State':
['Tamilnad','Kerala','WestBengal']}
df = pd.DataFrame(data)
print("DataFrame is\n", df)
df1 = df[df['City'].str.startswith('K') & df['State'].str.startswith('K')]
df1.to_csv('test.csv')

출력

City,State
Kochi,Kerala