Computer >> Hướng Dẫn Máy Tính >  >> Lập Trình >> Python

Lọc City/State bắt đầu bằng 'K' lưu ra CSV Python pandas

Giả sử bạn có DataFrame cột City, State và muốn lọc các hàng mà CẢ HAI cột đều bắt đầu bằng chữ 'K', sau đó lưu ra file CSV.

Giải pháp

  • Tạo DataFrame.
  • Lọc: df[df['City'].str.startswith('K') & df['State'].str.startswith('K')].
  • Xuất CSV: df_filtered.to_csv('test.csv', index=False).

Mã ví dụ

import pandas as pd

data = {
    'City': ['Chennai', 'Kochi', 'Kolkata'],
    'State': ['Tamilnad', 'Kerala', 'WestBengal']
}
df = pd.DataFrame(data)

print("DataFrame:\n", df)

df_filtered = df[df['City'].str.startswith('K') & df['State'].str.startswith('K')]
print("\nKết quả lọc:\n", df_filtered)

df_filtered.to_csv('test.csv', index=False)
print("\nĐã lưu ra test.csv")

Kết quả

DataFrame:
       City      State
0  Chennai   Tamilnad
1    Kochi     Kerala
2  Kolkata  WestBengal

Kết quả lọc:
   City  State
1  Kochi  Kerala

Đã lưu ra test.csv

Nội dung file test.csv:

City,State
Kochi,Kerala

Lưu ý:

  • str.startswith('K') phân biệt chữ hoa/thường. Dùng str.lower().str.startswith('k') nếu không phân biệt.
  • & là AND bitwise (cần dấu ngoặc đơn cho mỗi điều kiện).
  • index=False không ghi index ra file.