Comment utiliser le module csv pour lire et écrire des fichiers CSV dans Python 3.x

Comment utiliser le module csv pour lire et écrire des fichiers CSV dans Python 3.x

Introduction :
CSV (Comma-Separated Values) est un format de fichier courant utilisé pour stocker des données tabulaires. En Python, les fichiers CSV peuvent être traités à l'aide du module csv, qui fournit des fonctions de lecture et d'écriture de fichiers CSV. Cet article explique comment utiliser le module csv pour lire et écrire des fichiers CSV dans Python 3.x et fournit des exemples de code.

1. Utilisez le module csv pour lire les fichiers CSV

Tout d'abord, nous devons importer le module csv :

import csv

Ensuite, nous pouvons utiliser la fonction csv.reader() pour lire le fichier CSV. Cette fonction doit transmettre un objet fichier. Nous pouvons utiliser la fonction open() pour ouvrir un fichier CSV et transmettre l'objet fichier en tant que paramètre à la fonction csv.reader().

Supposons que nous ayons un fichier CSV appelé data.csv avec le contenu suivant :

name,age,city
John,25,New York
Peter,30,Chicago
Sarah,28,Los Angeles

Voici un exemple de code simple montrant comment utiliser le module csv pour lire un fichier CSV :

import csv

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

Exécutez le code ci-dessus, chaque ligne de les données du fichier CSV seront sorties. Dans cet exemple, nous utilisons une boucle for pour parcourir l'objet csv_reader afin d'imprimer les données du fichier CSV ligne par ligne.

2. Utilisez le module csv pour écrire des fichiers CSV

Semblable à la lecture de fichiers CSV, nous pouvons également utiliser le module csv pour écrire des fichiers CSV. Vous devez également importer le module csv :

import csv

Ensuite, nous pouvons utiliser la fonction csv.writer() pour créer un objet csv_writer, qui peut écrire des données dans le fichier CSV. Cette fonction doit transmettre un objet fichier. Vous pouvez également utiliser la fonction open() pour ouvrir un fichier CSV et transmettre l'objet fichier en tant que paramètre à la fonction csv.writer().

Ce qui suit est un exemple de code simple montrant comment utiliser le module csv pour écrire un fichier CSV :

import csv

data = [
    ['name', 'age', 'city'],
    ['John', '25', 'New York'],
    ['Peter', '30', 'Chicago'],
    ['Sarah', '28', 'Los Angeles']
]

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

Exécutez le code ci-dessus, un fichier CSV nommé output.csv sera créé dans le répertoire actuel et les données dans Data est écrit dans ce fichier.

Dans l'exemple de code, nous définissons d'abord une liste de données bidimensionnelle, qui contient chaque ligne de données dans le fichier CSV. Nous utilisons ensuite la fonction csv.writerows() pour écrire la liste complète dans un fichier CSV.

Résumé :
Avec le module csv, nous pouvons facilement lire et écrire des fichiers CSV dans Python 3.x. Lors de la lecture d'un fichier CSV, nous pouvons utiliser la fonction csv.reader() et la boucle for pour traiter les données ligne par ligne ; lors de l'écriture d'un fichier CSV, nous pouvons utiliser la fonction csv.writer() et la fonction csv_writer.writerows(). à Les données sont écrites dans un fichier CSV.

J'espère que cet article vous aidera à utiliser le module csv en Python pour lire et écrire des fichiers CSV. Bonnes études !

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!