Citirea și scrierea fișierelor CSV în Python

Fișierele CSV (Comma-Separated Values) sunt utilizate pe scară largă pentru stocarea și schimbul de date. În Python, puteți citi și scrie cu ușurință fișiere CSV folosind modulul încorporat csv. Acest articol vă va ghida prin elementele de bază ale lucrului cu fișierele CSV în Python.

Citirea fișierelor CSV

Pentru a citi un fișier CSV, puteți utiliza clasa csv.reader din modulul csv. Iată un exemplu simplu:

import csv

# Open the CSV file
with open('example.csv', mode='r') as file:
    reader = csv.reader(file)
    
    # Iterate over each row in the CSV
    for row in reader:
        print(row)

În acest exemplu, example.csv este deschis în modul de citire, iar csv.reader este folosit pentru a analiza fișierul. Fiecare rând al fișierului CSV este tipărit ca o listă de valori.

Scrierea fișierelor CSV

Pentru a scrie date într-un fișier CSV, utilizați clasa csv.writer. Iată cum puteți scrie o listă de rânduri într-un fișier CSV:

import csv

# Data to be written to the CSV
data = [
    ['Name', 'Age', 'City'],
    ['Alice', 30, 'New York'],
    ['Bob', 25, 'Los Angeles'],
    ['Charlie', 35, 'Chicago']
]

# Open the CSV file in write mode
with open('output.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    
    # Write each row to the CSV file
    writer.writerows(data)

În acest exemplu, output.csv este creat sau suprascris cu datele specificate. Metoda writer.writerows() este folosită pentru a scrie mai multe rânduri simultan.

Concluzie

Lucrul cu fișiere CSV în Python este simplu folosind modulul csv. Puteți citi și scrie cu ușurință fișiere CSV, făcând ușoară gestionarea datelor în acest format comun.