Файл чытання Python CSV

У гэтым паведамленні мы разгледзім, як чытаць і пісаць CSV-файлы на Python. У прыкладах выкарыстоўваецца модуль CSV і Pandas.



Python чытае файл CSV з дапамогай модуля CSV

У гэтым прыкладзе кода чытаецца файл з імем orders.csv і цыкл праз дадзеныя:

import csv f = open('orders.csv', 'rt') orders = csv.reader(f) for order in orders:
print(order) f.close()

Выхад заказаў.csv:


['OrderID', 'CustomerID', 'OrderDate'] ['10248', '4', '10/02/2020'] ['10249', '2', '10/02/2020'] ['10250', '7', '10/02/2020']

Чытайце файл CSV як слоўнік

Выкарыстанне DictReader метад кожны радок у файле CSV прадстаўлены ў выглядзе радка ў слоўніку, а загаловак слупка з'яўляецца ключом.

import csv orders = csv.DictReader(open('orders.csv')) for order in orders:
print(order)

Выхад заказаў.csv:


{'OrderID': '10248', 'CustomerID': '4', 'OrderDate': '10/02/2020'} {'OrderID': '10249', 'CustomerID': '2', 'OrderDate': '10/02/2020'} {'OrderID': '10250', 'CustomerID': '7', 'OrderDate': '10/02/2020'}

Прачытайце файл CSV з дапамогай Pandas

Каб выкарыстоўваць Pandas, нам спачатку трэба ўсталяваць бібліятэку Pandas.

Каб усталяваць, выканайце наступную каманду: pip3 install pandas.

import pandas orders = pandas.read_csv('orders.csv') print(orders)

Выхад заказаў.csv:

OrderID CustomerID OrderDate 10248

4
10/02/2020 10249

2
10/02/2020 10250

7
10/02/2020


Python Запіс CSV з выкарыстаннем Pandas

from pandas import DataFrame import pandas as pd order = pd.DataFrame({'OrderID': ['10251', '10252', '10253'],



'CustomerID': ['5', '1', '8'],



'OrderDate': ['11/02/2020', '11/02/2020', '11/02/2020']}) order.to_csv('newOrders.csv', index=False)

Выхад newOrders.csv:


OrderID,CustomerID,OrderDate 10251,5,11/02/2020 10252,1,11/02/2020 10253,8,11/02/2020

Дадаць дадзеныя ў існуючы файл CSV

Рэжым па змаўчанні пры запісе ў файл CSV - 'w'. Калі мы хочам дадаць дадзеныя ў існуючы файл CSV, мы павінны выкарыстоўваць рэжым дадання, напрыклад mode='a'

from pandas import DataFrame import pandas as pd order = pd.DataFrame({'OrderID': ['10254'],



'CustomerID': ['3'],



'OrderDate': ['11/02/2020']}) order.to_csv('newOrders.csv', mode='a', index=False, header=False)

Выхад newOrders.csv:

OrderID,CustomerID,OrderDate 10251,5,11/02/2020 10252,1,11/02/2020 10253,8,11/02/2020 10254,3,11/02/2020

Python Запіс CSV з выкарыстаннем модуля CSV

Н.Б. Першы радок выкарыстоўваецца ў якасці загалоўка слупка

import csv with open('orders.csv', 'w', newline='') as file:
order = csv.writer(file)
order.writerow(['OrderID', 'CustomerID', 'OrderDate'])
order.writerow(['10251', '6', '11/02/2020'])
order.writerow(['10252', '9', '11/02/2020'])
order.writerow(['10253', '5', '11/02/2020'])

Выхад заказаў.csv:


OrderID,CustomerID,OrderDate 10251,6,11/02/2020 10252,9,11/02/2020 10253,5,11/02/2020

Калі вы хочаце дадаць да існуючага файла csv з дапамогай модуля csv, вам трэба перадаць 'a' параметр да open() метад. Таксама трэба прапусціць 'загалоўкі'.

Напрыклад:

with open('orders.csv', 'a', newline='') as file:
order = csv.writer(file)
order.writerow(['10251', '6', '11/02/2020'])


Запіс CSV-файла з дапамогай DictWriter

Мы таксама можам выкарыстоўваць DictWriter метад csv клас для стварэння і запісу файла CSV.

import csv with open('orders.csv', 'w', newline='') as file:
fieldnames = ['OrderID', 'CustomerID', 'OrderDate']
order = csv.DictWriter(file, fieldnames=fieldnames)
order.writeheader()
order.writerow({'OrderID': '10251', 'CustomerID': 7, 'OrderDate': '11/02/2020'})
order.writerow({'OrderID': '10252', 'CustomerID': 3, 'OrderDate': '11/02/2020'})
order.writerow({'OrderID': '10253', 'CustomerID': 1, 'OrderDate': '11/02/2020'})

Выхад заказаў.csv:


OrderID,CustomerID,OrderDate 10251,7,11/02/2020 10252,3,11/02/2020 10253,1,11/02/2020