Чтение и запись CSV-файлов на Python

Файлы CSV (Comma-Separated Values) широко используются для хранения и обмена данными. В 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-файлы, что упрощает обработку данных в этом распространенном формате.