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

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

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

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

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

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

Для чтения 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-файлы

Для записи в 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-файлов.
• Не обрабатывать отсутствующие или некорректно сформированные данные в файле.

Что мы узнали?

В этом уроке вы научились:

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

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