Bagaimana untuk Mengecualikan Pengepala Semasa Memproses Fail CSV dengan Python?

Mengecualikan Pengepala Semasa Memproses Fail CSV dengan Python

Semasa memproses fail CSV, pengepala mungkin perlu dikecualikan daripada operasi tertentu. Ini boleh dicapai dengan menggunakan modul pembaca dan penulis CSV Python.

Dalam kod yang disediakan, melangkau baris pengepala diingini. Daripada memulakan pembolehubah baris kepada 1, pendekatan yang lebih mudah ialah melangkau baris pertama sebelum memproses yang lain. Ini boleh dicapai seperti berikut:

<code class="python">with open("tmob_notcleaned.csv", "rb") as infile, open("tmob_cleaned.csv", "wb") as outfile:
    reader = csv.reader(infile)
    next(reader, None)  # Skip the headers
    writer = csv.writer(outfile)
    for row in reader:
        # Process each row
        writer.writerow(row)</code>

Dengan memanggil seterusnya(pembaca, Tiada), baris pertama fail CSV diambil dan dibuang. Baris berikut kemudiannya boleh diproses dan ditulis ke fail output.

Selain itu, kod boleh dipermudahkan dengan menggunakan pengurus konteks untuk mengendalikan pembukaan dan penutupan fail secara automatik:

<code class="python">with open("tmob_notcleaned.csv", "rb") as infile, open("tmob_cleaned.csv", "wb") as outfile:
    reader = csv.reader(infile)
    writer = csv.writer(outfile)
    # Skip the headers
    if headers := next(reader, None):
        writer.writerow(headers)</code>

Dalam ini kod, pembolehubah pengepala pilihan menerima pengepala yang dilangkau, membenarkan pengguna menulisnya tanpa diproses ke fail output jika dikehendaki.

Atas ialah kandungan terperinci Bagaimana untuk Mengecualikan Pengepala Semasa Memproses Fail CSV dengan Python?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!