Введение в работу с CSV-файлами

Файлы CSV (значения, разделённые запятыми) — распространённый формат хранения табличных данных. Python предоставляет модуль csv для эффективного выполнения операций read и write с CSV-файлами и их обработки.

В этом уроке Вы узнаете, как работать с CSV-файлами с помощью встроенного модуля Python csv.

Что такое CSV-файлы

CSV-файлы — это текстовые файлы, в которых для разделения значений используется разделитель (обычно запятая). Каждая строка представляет строку таблицы, а каждое значение в строке — столбец.

Чтение CSV-файлов

Чтобы выполнить read CSV-файла, используйте функцию csv.reader. Эта функция считывает каждую строку как список значений:

# Reading a CSV file
import csv

with open("data.csv", "r") as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)  # Each row is a list of values

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

Чтобы выполнить write в CSV-файл, используйте функцию csv.writer. Эта функция позволяет записывать строки в CSV-файл:

# Writing to a CSV file
import csv

with open("output.csv", "w", newline="") as file:
    writer = csv.writer(file)
    writer.writerow(["Name", "Age", "City"])
    writer.writerow(["Alice", 30, "New York"])
    writer.writerow(["Bob", 25, "London"])

Чтение CSV-файлов с заголовками

Используйте csv.DictReader, чтобы считывать CSV-файлы в словари, где ключами являются заголовки столбцов:

# Reading a CSV file with headers
import csv

with open("data.csv", "r") as file:
    reader = csv.DictReader(file)
    for row in reader:
        print(row)  # Each row is a dictionary with column headers as keys

Запись CSV-файлов с заголовками

Используйте csv.DictWriter, чтобы записывать словари в CSV-файлы с заголовками:

# Writing a CSV file with headers
import csv

with open("output.csv", "w", newline="") as file:
    fieldnames = ["Name", "Age", "City"]
    writer = csv.DictWriter(file, fieldnames=fieldnames)

    writer.writeheader()  # Write the header row
    writer.writerow({"Name": "Alice", "Age": 30, "City": "New York"})
    writer.writerow({"Name": "Bob", "Age": 25, "City": "London"})

Пользовательские разделители в CSV-файлах

При чтении или записи CSV-файлов можно указать пользовательский разделитель. Например, в файлах с разделением табуляцией в качестве разделителя используется табуляция (\t):

# Using a custom delimiter
import csv

with open("data.tsv", "r") as file:
    reader = csv.reader(file, delimiter="\t")
    for row in reader:
        print(row)  # Each row is split by tab

Работа с большими CSV-файлами

При работе с большими CSV-файлами используйте цикл для поэтапной обработки строк, не загружая весь файл в память:

# Reading a large CSV file efficiently
import csv

with open("large_data.csv", "r") as file:
    reader = csv.reader(file)
    for row in reader:
        # Process each row
        print(row)

Распространённые ошибки при работе с CSV-файлами

Вот некоторые ошибки, которых следует избегать:

• Забывать использовать newline="" при записи CSV-файлов в Windows.
• Использовать неправильные разделители для нестандартных CSV-файлов.
• Не обрабатывать отсутствующие или некорректные данные в файле.

Что мы узнали

В этом уроке Вы узнали:

• Как выполнять операции read и write с CSV-файлами с помощью модуля Python csv.
• Как работать с CSV-файлами, содержащими заголовки, используя DictReader и DictWriter.
• Как эффективно работать с большими CSV-файлами.

Отличная работа! Перейдём к следующей теме.