Читання та запис файлів CSV на Python

Файли CSV (значення, розділені комами) широко використовуються для зберігання та обміну даними. У Python ви можете легко читати та записувати файли CSV за допомогою вбудованого модуля csv. Ця стаття проведе вас через основи роботи з файлами CSV у Python.

Читання файлів CSV

Щоб прочитати файл CSV, ви можете використовувати клас csv.reader з модуля csv. Ось простий приклад:

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)

У цьому прикладі example.csv відкривається в режимі читання, а csv.reader використовується для аналізу файлу. Кожен рядок файлу CSV друкується як список значень.

Запис файлів CSV

Щоб записати дані у файл CSV, використовуйте клас csv.writer. Ось як можна записати список рядків у файл 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)

У цьому прикладі output.csv створюється або перезаписується вказаними даними. Метод writer.writerows() використовується для запису кількох рядків одночасно.

Висновок

Робота з файлами CSV у Python проста за допомогою модуля csv. Ви можете легко читати та записувати файли CSV, що спрощує обробку даних у цьому поширеному форматі.