Python CSV: работа с CSV-файлами

Python csv - это модуль языка программирования Python, который предоставляет возможности для работы с CSV-файлами. CSV (Comma-Separated Values) представляет собой текстовый файл, в котором значения разделены запятыми. CSV-файлы широко применяются для обмена данными между различными приложениями, такими как электронные таблицы, базы данных и другие системы.

Для работы с CSV в Python используется модуль csv, который предоставляет классы и методы для чтения, записи и обработки данных в формате CSV. Давайте рассмотрим некоторые основные операции с CSV в Python с помощью модуля csv.

Для начала работы с CSV-файлом мы должны импортировать модуль csv:

import csv

Чтение CSV-файла:

Для чтения CSV-файла мы можем использовать метод reader() модуля csv. Метод reader() принимает в качестве параметра объект файла и возвращает объект-итератор, который можно использовать для итерации по значениям в файле.

Пример чтения CSV-файла:

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

В коде выше открываем файл 'file.csv' в режиме чтения ('r') и создаем объект csv_reader с помощью метода reader(). Затем мы проходимся по каждой строке (row) в файле и выводим ее на печать.

Запись в CSV-файл:

Для записи данных в CSV-файл мы используем метод writer() модуля csv. Метод writer() также принимает в качестве параметра объект файла и возвращает объект csv.writer, который мы можем использовать для записи данных в файл.

Пример записи в CSV-файл:

data = [
    ['Name', 'Age', 'City'],
    ['John', '25', 'New York'],
    ['Alice', '30', 'London']
]

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

В приведенном коде мы создаем список data, который содержит данные, которые мы хотим записать в CSV-файл. Затем мы открываем файл 'file.csv' в режиме записи ('w') и создаем объект csv_writer с помощью метода writer(). Затем мы используем метод writerows() для записи всех строк из списка data в CSV-файл.

Обработка данных в CSV-файле:

Модуль csv также предоставляет дополнительные возможности для обработки данных в CSV-файлах. Например, мы можем использовать метод DictReader для чтения данных в формате словаря, где первая строка CSV-файла используется в качестве заголовков столбцов.

Пример чтения CSV-файла с использованием DictReader:

with open('file.csv', 'r') as csv_file:
    csv_reader = csv.DictReader(csv_file)
    for row in csv_reader:
        print(row['Name'], row['Age'], row['City'])

В приведенном коде мы создаем объект csv_reader с помощью метода DictReader(), что позволяет использовать заголовки столбцов в качестве ключей словаря. Затем мы проходим по каждой строке в CSV-файле и выводим значения по ключам 'Name', 'Age' и 'City' на печать.

Это лишь небольшой набор возможностей, которые предоставляет модуль csv в Python. Используя этот модуль, вы можете выполнять различные операции с CSV-файлами, такие как чтение, запись, обработка данных и другие.