Cara menggunakan modul csv untuk membaca dan menulis fail CSV dalam Python 3.x

Cara menggunakan modul csv untuk membaca dan menulis fail CSV dalam Python 3.x

Pengenalan:
CSV (Nilai Dipisahkan Koma) ialah format fail biasa yang digunakan untuk menyimpan data jadual. Dalam Python, fail CSV boleh diproses menggunakan modul csv, yang menyediakan fungsi untuk membaca dan menulis fail CSV. Artikel ini menerangkan cara menggunakan modul csv untuk membaca dan menulis fail CSV dalam Python 3.x dan menyediakan contoh kod.

1. Gunakan modul csv untuk membaca fail CSV

Pertama, kita perlu mengimport modul csv:

import csv

Seterusnya, kita boleh menggunakan fungsi csv.reader() untuk membaca fail CSV. Fungsi ini perlu lulus dalam objek fail Kita boleh menggunakan fungsi open() untuk membuka fail CSV dan menghantar objek fail sebagai parameter kepada fungsi csv.reader().

Andaikan kita mempunyai fail CSV bernama data.csv dengan kandungan berikut:

name,age,city
John,25,New York
Peter,30,Chicago
Sarah,28,Los Angeles

Berikut ialah contoh kod ringkas yang menunjukkan cara menggunakan modul csv untuk membaca fail CSV:

import csv

with open('data.csv', 'r') as file:
    csv_reader = csv.reader(file)
    for row in csv_reader:
        print(row)

Jalankan kod di atas, Setiap baris data dalam fail CSV akan dikeluarkan. Dalam contoh ini, kami menggunakan gelung for untuk lelaran pada objek csv_reader untuk mencetak data dalam fail CSV baris demi baris.

2. Gunakan modul csv untuk menulis fail CSV

Sama seperti membaca fail CSV, kami juga boleh menggunakan modul csv untuk menulis fail CSV. Anda juga perlu mengimport modul csv:

import csv

Kemudian, kita boleh menggunakan fungsi csv.writer() untuk mencipta objek csv_writer, yang boleh menulis data ke fail CSV. Fungsi ini perlu menghantar objek fail Anda juga boleh menggunakan fungsi open() untuk membuka fail CSV dan menghantar objek fail sebagai parameter kepada fungsi csv.writer().

Berikut ialah contoh kod ringkas yang menunjukkan cara menggunakan modul csv untuk menulis fail CSV:

import csv

data = [
    ['name', 'age', 'city'],
    ['John', '25', 'New York'],
    ['Peter', '30', 'Chicago'],
    ['Sarah', '28', 'Los Angeles']
]

with open('output.csv', 'w', newline='') as file:
    csv_writer = csv.writer(file)
    csv_writer.writerows(data)

Jalankan kod di atas, fail CSV bernama output.csv akan dibuat dalam direktori semasa dan data dalam Data ditulis pada fail ini.

Dalam contoh kod, kami mula-mula menentukan data senarai dua dimensi, yang mengandungi setiap baris data dalam fail CSV. Kami kemudian menggunakan fungsi csv.writerows() untuk menulis keseluruhan senarai ke fail CSV.

Ringkasan:
Dengan modul csv, kami boleh membaca dan menulis fail CSV dengan mudah dalam Python 3.x. Apabila membaca fail CSV, kita boleh menggunakan fungsi csv.reader() dan untuk gelung memproses baris data demi baris apabila menulis fail CSV, kita boleh menggunakan fungsi csv.writer() dan fungsi csv_writer.writerows(); kepada Data ditulis ke dalam fail CSV.

Semoga artikel ini dapat membantu anda menggunakan modul csv dalam Python untuk membaca dan menulis fail CSV. Selamat belajar!

Atas ialah kandungan terperinci Cara menggunakan modul csv untuk membaca dan menulis fail CSV dalam Python 3.x. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!