Курс Python → Работа с CSV файлами в Python

При работе с базами данных в Python, хранение данных в виде CSV файлов является распространенным способом. Однако, для удобства чтения и записи данных, рекомендуется использовать методы DictReader и DictWriter из модуля csv. С их помощью можно упростить работу с данными и сделать код более понятным для коллег.

Метод DictReader позволяет читать данные из CSV файла и возвращать словарь для каждой строки, где ключами являются имена столбцов. Это значительно упрощает доступ к данным по ключу, вместо индекса столбца. Пример использования DictReader показан на картинке.

import csv

with open('data.csv', 'r') as file:
    reader = csv.DictReader(file)
    for row in reader:
        print(row['column_name'])

Для записи данных в CSV файл с использованием DictWriter, необходимо создать объект writer и вызвать метод writeheader для записи первой строки с именами столбцов. Затем, данные могут быть записаны с помощью метода writerow, где аргументом является словарь с данными для каждой строки.

import csv

data = [{'name': 'Alice', 'age': 25}, {'name': 'Bob', 'age': 30}]

with open('data.csv', 'w', newline='') as file:
    columns = ['name', 'age']
    writer = csv.DictWriter(file, fieldnames=columns)
    writer.writeheader()
    for row in data:
        writer.writerow(row)

Использование DictReader и DictWriter при работе с базами данных в Python позволяет упростить код, делая его более читаемым и понятным. Эти методы помогут не только вам, но и вашим коллегам, которые будут работать с вашим кодом. Помните, что хорошо структурированный и понятный код — это залог успешного сотрудничества и разработки.

Твои коллеги будут рады, поделись в