Как использовать Python для работы с CSV

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

Освойте Python на курсе от Skypro. Вас ждут 400 часов обучения и практики (достаточно десяти часов в неделю), подготовка проектов для портфолио, индивидуальная проверка домашних заданий и помощь опытных наставников. Получится, даже если у вас нет опыта в IT.

Библиотека csv

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

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

Для чтения CSV-файлов сначала импортируйте библиотеку csv и используйте функцию csv.reader():

import csv

with open('example.csv', 'r') as csvfile:
    csvreader = csv.reader(csvfile)
    for row in csvreader:
        print(row)

📝 В этом примере мы открываем файл example.csv на чтение и передаем его функции csv.reader(). Затем мы проходимся по каждой строке и выводим ее.

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

Для записи данных в CSV-файл используйте функцию csv.writer():

import csv

data = [['Name', 'Age', 'City'],
        ['Alice', '30', 'New York'],
        ['Bob', '25', 'San Francisco']]

with open('output.csv', 'w') as csvfile:
    csvwriter = csv.writer(csvfile)
    for row in data:
        csvwriter.writerow(row)

📝 В этом примере мы создаем список списков data, который содержит информацию для записи в CSV-файл. Затем мы открываем файл output.csv на запись и передаем его функции csv.writer(). Далее мы проходимся по каждой строке и записываем ее в файл.

Работа с CSV-файлами с использованием Pandas

Pandas — это популярная библиотека Python для работы с табличными данными. Она предоставляет удобные функции для чтения и записи CSV-файлов. Для начала установите библиотеку Pandas при помощи pip:

pip install pandas

Чтение CSV-файлов с использованием Pandas

Чтобы прочитать CSV-файл с использованием Pandas, используйте функцию pandas.read_csv():

import pandas as pd

data = pd.read_csv('example.csv')
print(data)

📝 В этом примере мы импортируем библиотеку Pandas и читаем данные из файла example.csv. Затем выводим их на экран.

На курсе Skypro «Python-разработчик» освоите основные инструменты программирования, получите опыт на реальных проектах и сможете стартовать в профессии уверенным новичком. Преподаватели — практикующие программисты с большим опытом, а в центре карьеры помогут составить цепляющее резюме и подготовиться к собеседованию.

Запись CSV-файлов с использованием Pandas

Для записи данных в CSV-файл с использованием Pandas используйте метод DataFrame.to_csv():

import pandas as pd

data = {'Name': ['Alice', 'Bob'],
        'Age': [30, 25],
        'City': ['New York', 'San Francisco']}

df = pd.DataFrame(data)
df.to_csv('output.csv', index=False)

📝 В этом примере мы создаем словарь data, который содержит информацию для записи в CSV-файл. Затем мы преобразуем словарь в объект DataFrame и записываем его в файл output.csv.

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