Membaca dan Menulis File CSV di Python

File CSV (Comma-Separated Values) banyak digunakan untuk penyimpanan dan pertukaran data. Dalam Python, Anda dapat dengan mudah membaca dan menulis ke file CSV menggunakan modul csv bawaan. Artikel ini akan memandu Anda mempelajari dasar-dasar bekerja dengan file CSV dalam Python.

Membaca File CSV

Untuk membaca file CSV, Anda dapat menggunakan kelas csv.reader dari modul csv. Berikut contoh sederhananya:

import csv

# Open the CSV file
with open('example.csv', mode='r') as file:
    reader = csv.reader(file)
    
    # Iterate over each row in the CSV
    for row in reader:
        print(row)

Dalam contoh ini, example.csv dibuka dalam mode baca, dan csv.reader digunakan untuk mengurai berkas. Setiap baris berkas CSV dicetak sebagai daftar nilai.

Menulis File CSV

Untuk menulis data ke file CSV, gunakan kelas csv.writer. Berikut cara menulis daftar baris ke file CSV:

import csv

# Data to be written to the CSV
data = [
    ['Name', 'Age', 'City'],
    ['Alice', 30, 'New York'],
    ['Bob', 25, 'Los Angeles'],
    ['Charlie', 35, 'Chicago']
]

# Open the CSV file in write mode
with open('output.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    
    # Write each row to the CSV file
    writer.writerows(data)

Dalam contoh ini, output.csv dibuat atau ditimpa dengan data yang ditentukan. Metode writer.writerows() digunakan untuk menulis beberapa baris sekaligus.

Kesimpulan

Bekerja dengan file CSV di Python mudah dilakukan dengan menggunakan modul csv. Anda dapat dengan mudah membaca dan menulis ke file CSV, sehingga memudahkan penanganan data dalam format umum ini.