Computer >> Máy Tính >  >> Lập trình >> Python

Viết chương trình bằng Python để lưu trữ tên thành phố và tiểu bang bắt đầu bằng ‘k’ trong DataFrame nhất định thành một tệp CSV mới

Đầu vào -

Giả sử, chúng ta có DataFrame với các cột Thành phố và Tiểu bang và tìm thành phố, tên tiểu bang bắt đầu bằng ‘k’ và lưu trữ vào một tệp CSV khác như được hiển thị bên dưới -

City,State
Kochi,Kerala

Giải pháp

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước dưới đây.

  • Xác định DataFrame

  • Kiểm tra thành phố bắt đầu bằng ‘k’ như được định nghĩa bên dưới,

df[df['City'].str.startswith('K') & df['State'].str.startswith('K')]
  • Cuối cùng, lưu trữ dữ liệu trong tệp ‘CSV’ như bên dưới,

df1.to_csv(‘test.csv’)

Ví dụ

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn.

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')

Đầu ra

City,State
Kochi,Kerala