Lecture et écriture de fichiers CSV en Python

Les fichiers CSV (Comma-Separated Values) sont largement utilisés pour le stockage et l'échange de données. En Python, vous pouvez facilement lire et écrire dans des fichiers CSV à l'aide du module intégré csv. Cet article vous guidera à travers les bases de l'utilisation des fichiers CSV en Python.

Lecture de fichiers CSV

Pour lire un fichier CSV, vous pouvez utiliser la classe csv.reader du module csv. Voici un exemple simple:

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)

Dans cet exemple, example.csv est ouvert en mode lecture et csv.reader est utilisé pour analyser le fichier. Chaque ligne du fichier CSV est imprimée sous forme de liste de valeurs.

Écriture de fichiers CSV

Pour écrire des données dans un fichier CSV, utilisez la classe csv.writer. Voici comment écrire une liste de lignes dans un fichier 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)

Dans cet exemple, output.csv est créé ou écrasé par les données spécifiées. La méthode writer.writerows() est utilisée pour écrire plusieurs lignes à la fois.

Conclusion

Travailler avec des fichiers CSV dans Python est simple grâce au module csv. Vous pouvez facilement lire et écrire dans des fichiers CSV, ce qui simplifie la gestion des données dans ce format courant.