Чтение и запись файлов CSV в Python: csv, pandas и NumPy

Для чтения и записи файлов CSV в Python вы можете использовать различные методы и библиотеки. Вот несколько часто используемых подходов:

  1. Использование модуля csv. Python предоставляет встроенный модуль csv, который предлагает функции для чтения и записи файлов CSV. Вот пример того, как его использовать:
import csv
# Reading a CSV file
with open('file.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)
# Writing to a CSV file
with open('file.csv', 'w') as file:
    writer = csv.writer(file)
    writer.writerow(['Name', 'Age', 'Country'])
    writer.writerow(['John', '25', 'USA'])
    writer.writerow(['Emma', '30', 'Canada'])
  1. Использование библиотеки pandas: Pandas — это мощная библиотека манипулирования данными на Python. Он обеспечивает удобный способ чтения и записи файлов CSV с помощью функций read_csv()и to_csv(). Вот пример:
import pandas as pd
# Reading a CSV file
df = pd.read_csv('file.csv')
print(df)
# Writing to a CSV file
data = {'Name': ['John', 'Emma'], 'Age': [25, 30], 'Country': ['USA', 'Canada']}
df = pd.DataFrame(data)
df.to_csv('file.csv', index=False)
  1. Использование библиотеки NumPy: NumPy — еще одна популярная библиотека для числовых вычислений на Python. Он предоставляет такие функции, как genfromtxt()и savext(), которые можно использовать для чтения и записи файлов CSV. Вот пример:
import numpy as np
# Reading a CSV file
data = np.genfromtxt('file.csv', delimiter=',')
print(data)
# Writing to a CSV file
data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
np.savetxt('file.csv', data, delimiter=',')