Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

เขียนโปรแกรมใน Python เพื่อเก็บชื่อเมืองและรัฐที่ขึ้นต้นด้วย 'k' ใน DataFrame ที่กำหนดลงในไฟล์ CSV ใหม่


ป้อนข้อมูล

สมมติว่า เรามี DataFrame ที่มีคอลัมน์ City และ State และค้นหาเมือง ชื่อรัฐขึ้นต้นด้วย '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