Comment importer un fichier CSV dans une liste en Python ?

Importer un fichier CSV dans une liste en Python

L'importation d'un fichier CSV dans une liste en Python est une tâche courante. Dans cet article, nous allons montrer comment y parvenir à l'aide du module csv.

Méthode :

1. Importer le module csv : importer csv
2. Ouvrez le fichier CSV à l'aide d'un gestionnaire de contexte : avec open('file.csv', newline='') comme f:
3. Créer un objet lecteur : reader = csv.reader(f)
4. Convertir l'objet lecteur en liste : data = list(reader)

Exemple :

Considérons un fichier CSV avec ce qui suit data :

This is the first line,Line1
This is the second line,Line2
This is the third line,Line3

Pour importer ces données dans une liste, on peut utiliser le code suivant :

import csv

with open('file.csv', newline='') as f:
    reader = csv.reader(f)
    data = list(reader)

print(data)

Sortie :

[['This is the first line', 'Line1'], ['This is the second line', 'Line2'], ['This is the third line', 'Line3']]

Remarque : Si vous avez besoin de tuples au lieu de listes, vous pouvez modifier le code ci-dessus comme suit :

with open('file.csv', newline='') as f:
    reader = csv.reader(f)
    data = [tuple(row) for row in reader]

Cela permettra produire une liste de tuples :

[('This is the first line', 'Line1'), ('This is the second line', 'Line2'), ('This is the third line', 'Line3')]

Pour les utilisateurs de Python 2, vous pouvez utiliser le code suivant :

import csv
with open('file.csv', 'rb') as f:
    reader = csv.reader(f)
    your_list = list(reader)

print your_list

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!