Exemplu de cititor CSV în Python: Cum să lucrezi cu fișiere CSV

Citirea unui fișier CSV de bază

Unul dintre cele mai simple exemple de citire a unui fișier CSV în Python este folosirea modulului csv și a funcției csv.reader(). Iată un exemplu de cod:

import csv

with open('exemplu.csv', mode='r') as file:
    csv_reader = csv.reader(file)
    for row in csv_reader:
        print(row)

În acest exemplu, deschidem fișierul exemplu.csv în modul citire (mode='r') și utilizăm funcția csv.reader() pentru a citi fișierul. Apoi, iterăm prin fiecare rând din fișier și îl afișăm. Rezultatul ar arăta astfel:

['Nume', 'Vârstă', 'Oraș']
['Ion', '30', 'București']
['Maria', '25', 'Cluj']
['Alex', '35', 'Timișoara']

Observați că fiecare rând din fișier este citit ca o listă de valori, iar valorile din fiecare rând sunt separate de virgulă.

Citirea unui fișier CSV cu antet

De obicei, fișierele CSV conțin un antet care descrie fiecare coloană. Dacă doriți să lucrați cu aceste fișiere și să le tratați corect, este recomandat să săriți peste primul rând (antetul). În acest caz, putem folosi funcția next() pentru a sări peste primul rând și a începe citirea datelor.

import csv

with open('exemplu.csv', mode='r') as file:
    csv_reader = csv.reader(file)
    next(csv_reader)  # Sărim peste antet
    for row in csv_reader:
        print(row)

Acum, la rularea codului, vom obține doar datele, fără antet:

['Ion', '30', 'București']
['Maria', '25', 'Cluj']
['Alex', '35', 'Timișoara']

Citirea unui fișier CSV folosind DictReader

Un alt mod de a citi un fișier CSV în Python este utilizând csv.DictReader(). Aceasta funcție citește fiecare rând ca un dicționar, unde cheia este numele coloanei (din antet), iar valoarea este valoarea corespunzătoare pentru fiecare rând. Iată un exemplu:

import csv

with open('exemplu.csv', mode='r') as file:
    csv_reader = csv.DictReader(file)
    for row in csv_reader:
        print(row)

Rezultatul va arăta astfel:

{'Nume': 'Ion', 'Vârstă': '30', 'Oraș': 'București'}
{'Nume': 'Maria', 'Vârstă': '25', 'Oraș': 'Cluj'}
{'Nume': 'Alex', 'Vârstă': '35', 'Oraș': 'Timișoara'}

După cum vedeți, fiecare rând este acum un dicționar, iar valorile sunt accesibile prin cheile corespunzătoare.

Exemplu de procesare a datelor CSV

Odată ce am învățat cum să citim un fișier CSV, haideți să vedem un exemplu de procesare a datelor. Să presupunem că avem un fișier CSV cu informații despre persoane și dorim să aflăm media vârstei acestora.

import csv

total_varste = 0
numar_persoane = 0

with open('persoane.csv', mode='r') as file:
    csv_reader = csv.DictReader(file)
    for row in csv_reader:
        total_varste += int(row['Vârstă'])
        numar_persoane += 1

media_varstei = total_varste / numar_persoane
print("Media vârstei este:", media_varstei)

În acest exemplu, citim fișierul CSV, adunăm toate vârstele și calculăm media acestora. Acest tip de procesare a datelor este foarte util atunci când lucrăm cu fișiere mari și trebuie să extragem informații utile din ele.

Scrierea într-un fișier CSV

Pe lângă citirea fișierelor CSV, Python vă permite să scrieți și să modificați fișiere CSV. Iată cum putem scrie date într-un fișier CSV:

import csv

date_de_scris = [
    ['Nume', 'Vârstă', 'Oraș'],
    ['Ion', '30', 'București'],
    ['Maria', '25', 'Cluj'],
    ['Alex', '35', 'Timișoara']
]

with open('output.csv', mode='w', newline='') as file:
    csv_writer = csv.writer(file)
    csv_writer.writerows(date_de_scris)

Acest cod va crea un fișier CSV nou numit output.csv și va scrie datele specificate în acesta. Observați că folosim newline='' pentru a preveni problemele de formatare în fișierul rezultat.

Concluzie

Python oferă un mod foarte simplu și eficient de a lucra cu fișiere CSV prin intermediul bibliotecii csv. Fie că doriți să citiți, să scrieți sau să procesați datele dintr-un fișier CSV, Python vă oferă toate instrumentele necesare pentru a face acest lucru ușor și rapid. Sperăm că acest articol v-a ajutat să înțelegeți cum să folosiți cititorul CSV în Python cu exemple clare și utile.