เขียนโปรแกรม Python กรองชื่อเมืองและจังหวัดที่ขึ้นต้นด้วย 'ก' บันทึกเป็น CSV

สมมติว่าคุณมี DataFrame ที่มีคอลัมน์ City และ State ต้องการกรองเฉพาะแถวที่ทั้งสองคอลัมน์ขึ้นต้นด้วย 'ก' (K) แล้วบันทึกเป็นไฟล์ CSV

วิธีแก้ปัญหา

1. สร้าง DataFrame
2. กรองด้วย df['City'].str.startswith('K') & df['State'].str.startswith('K')
3. บันทึกเป็น CSV ด้วย df.to_csv('test.csv')

ตัวอย่างโค้ด

import pandas as pd

data = {'City': ['Chennai', 'Kochi', 'Kolkata'],
        'State': ['Tamilnad', 'Kerala', 'WestBengal']}

df = pd.DataFrame(data)
print("DataFrame:\n", df)

# กรองเมืองและจังหวัดที่ขึ้นต้นด้วย 'K' (ตัวใหญ่)
df_filtered = df[df['City'].str.startswith('K') & df['State'].str.startswith('K')]

print("\nข้อมูลที่กรองได้:")
print(df_filtered)

# บันทึกเป็น CSV
df_filtered.to_csv('test.csv', index=False)
print("\nบันทึกเป็น test.csv เรียบร้อย")

ผลลัพธ์ (test.csv)

City,State
Kochi,Kerala

คำอธิบาย:

• str.startswith('K') ตรวจสอบว่า string เริ่มต้นด้วย 'K' (case-sensitive)
• ใช้ & รวมเงื่อนไขทั้งสอง (City และ State ต้องขึ้นต้นด้วย K ทั้งคู่)
• index=False ใน to_csv() ไม่บันทึก index ลงไฟล์
• ถ้าต้องการ case-insensitive: str.lower().str.startswith('k')