Python и CSV: как работать с файлами CSV в Python
Python CSV - это модуль Python, который позволяет работать с файловыми форматами CSV (Comma Separated Values), которые широко используются для представления табличных данных.
Чтобы начать использование CSV в Python, сначала нужно импортировать модуль CSV:
python
import csv
Модуль CSV имеет несколько методов для работы с CSV файлами:
### 1. Чтение CSV файлов
Для чтения CSV файлов в Python используйте метод reader(), который создает объект вашего CSV файла. Затем можно использовать цикл for для чтения каждой строки CSV файла.
Вот пример:
python
import csv
with open('example.csv', mode='r') as csvfile:
reader = csv.reader(csvfile)
for row in reader:
print(row)
В этом примере файл example.csv открывается для чтения, все строки CSV файла считываются в переменную reader, и затем выводятся каждая строка в консоль.
### 2. Запись CSV файлов
Вы можете использовать метод writer() для записи CSV-файлов. Он создает объект записи, который вы можете использовать для записи строк CSV в файл.
Вот пример записи двух строк в файл example.csv:
python
import csv
with open('example.csv', mode='w') as csvfile:
writer = csv.writer(csvfile)
writer.writerow(['Name', 'Age', 'Gender'])
writer.writerow(['John', '25', 'Male'])
В этом примере файл example.csv создается для записи, создается объект записи, состоящий из трех колонок, и затем две строки данных сохраняются в файл.
### 3. Использование CSV файлов в виде словарей
Модуль CSV Python также поддерживает чтение и запись CSV файлов в виде словарей, что позволяет более удобно работать с данными.
Для чтения CSV файла как словаря в Python можно использовать модуль DictReader.
Вот пример:
python
import csv
with open('example.csv') as csvfile:
reader = csv.DictReader(csvfile)
for row in reader:
print(row['Name'], row['Age'], row['Gender'])
### 4. Запись CSV файлов в виде словарей
Модуль CSV также поддерживает запись CSV файлов в виде словаря с помощью объекта DictWriter Python.
Вот пример:
python
import csv
with open('example.csv', mode='w') as csvfile:
fieldnames = ['Name', 'Age', 'Gender']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writeheader()
writer.writerow({'Name': 'John', 'Age': '25', 'Gender': 'Male'})
writer.writerow({'Name': 'Jenny', 'Age': '32', 'Gender': 'Female'})
В этом примере файл example.csv создается для записи, задаются имена столбцов (fieldnames), создается объект записи словаря с именами столбцов, записывается заголовок, и затем две строки данных с сохранением в файл.
Надеемся, что эта информация будет полезна при работе с CSV файлами в Python.