- À propos de la recherche d'une valeur dans un fichier CSV en Python
- Préparer le fichier CSV utilisé pour la recherche
- Rechercher une valeur avec csv.reader()
- Rechercher une valeur dans une colonne précise avec csv.DictReader()
- Rechercher plusieurs occurrences d'une même valeur
- Effectuer une recherche partielle et ignorer la casse
- Rechercher avec plusieurs critères
- Rechercher des valeurs numériques dans un fichier CSV
- Créer une fonction réutilisable de recherche dans un fichier CSV
- À retenir !
1. À propos de la recherche d'une valeur dans un fichier CSV en Python
La recherche d'une valeur dans un fichier CSV est une opération courante lorsqu'un programme doit retrouver un étudiant, un produit, une ville, un identifiant ou toute autre information enregistrée dans un fichier tabulaire.
Le module standard csv de Python permet de parcourir les lignes d'un fichier CSV et de tester les valeurs rencontrées. Nous pouvons utiliser csv.reader() pour travailler avec les positions des colonnes ou csv.DictReader() pour rechercher une valeur à partir du nom d'une colonne.
Le principe général consiste à parcourir les lignes et à comparer la valeur recherchée avec les données du fichier. L'exemple suivant illustre ce principe sur une simple liste Python.
1 2 3 4 5 6 7 | noms = ["Amine", "Sara", "Yassine", "Lina"] recherche = "Sara" for nom in noms: if nom == recherche: print("Valeur trouvée :", nom) break |
Sortie :
1 | Valeur trouvée : Sara |
Nous allons appliquer le même principe aux lignes d'un fichier CSV, puis étudier des recherches plus avancées : recherche dans une colonne précise, plusieurs résultats, recherche partielle, recherche insensible à la casse et combinaison de plusieurs critères.
2. Préparer le fichier CSV utilisé pour la recherche
Considérons un fichier nommé etudiants.csv. Il contient les colonnes id, nom, age, ville, note et filiere.
Voici le contenu utilisé dans les exemples de ce tutoriel.
1 2 3 4 5 6 7 | id,nom,age,ville,note,filiere 1,Amine,21,Oujda,15.5,Mathématiques 2,Sara,20,Rabat,17,Informatique 3,Yassine,22,Fès,13.5,Physique 4,Lina,19,Casablanca,18,Informatique 5,Omar,21,Oujda,14,Mathématiques 6,Nadia,23,Rabat,16,Informatique |
Sortie :
1 | Le fichier contient 6 colonnes et 6 lignes de données. |
Nous pouvons commencer par afficher le fichier afin de vérifier les données dans lesquelles les recherches seront effectuées.
1 2 | with open("etudiants.csv", "r", encoding="utf-8") as fichier: print(fichier.read()) |
Sortie :
1 2 3 4 5 6 7 | id,nom,age,ville,note,filiere 1,Amine,21,Oujda,15.5,Mathématiques 2,Sara,20,Rabat,17,Informatique 3,Yassine,22,Fès,13.5,Physique 4,Lina,19,Casablanca,18,Informatique 5,Omar,21,Oujda,14,Mathématiques 6,Nadia,23,Rabat,16,Informatique |
Dans les sections suivantes, nous rechercherons différentes valeurs dans ce fichier en utilisant csv.reader() puis csv.DictReader().
3. Rechercher une valeur avec csv.reader()
La fonction csv.reader() retourne chaque ligne du fichier sous forme de liste. Nous pouvons donc rechercher une valeur en utilisant l'opérateur in pour vérifier si elle apparaît dans une ligne.
Le programme suivant recherche la valeur "Yassine" dans toutes les lignes du fichier.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | import csv valeur_recherchee = "Yassine" trouve = False with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.reader(fichier) for ligne in lecteur: if valeur_recherchee in ligne: print("Valeur trouvée :", ligne) trouve = True break if not trouve: print("Valeur introuvable.") |
Sortie :
1 | Valeur trouvée : ['3', 'Yassine', '22', 'Fès', '13.5', 'Physique'] |
L'instruction break arrête immédiatement la boucle après la première correspondance. Cette technique est intéressante lorsque nous souhaitons uniquement savoir si une valeur existe.
Nous pouvons également rechercher dans une colonne particulière en utilisant son indice. Dans notre fichier, le nom se trouve à l'indice 1.
1 2 3 4 5 6 7 8 9 10 11 12 | import csv nom_recherche = "Lina" with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.reader(fichier) next(lecteur, None) for ligne in lecteur: if ligne[1] == nom_recherche: print("Étudiante trouvée :", ligne) break |
Sortie :
1 | Étudiante trouvée : ['4', 'Lina', '19', 'Casablanca', '18', 'Informatique'] |
Cette méthode fonctionne correctement, mais elle suppose de connaître la position exacte de la colonne. Pour des fichiers possédant une ligne d'en-tête, DictReader() permet d'utiliser directement les noms des colonnes.
4. Rechercher une valeur dans une colonne précise avec csv.DictReader()
La classe csv.DictReader() représente chaque ligne sous forme de dictionnaire. Nous pouvons donc effectuer une recherche dans une colonne précise avec une expression comme ligne["nom"] ou ligne["ville"].
Le programme suivant recherche l'étudiante nommée Sara dans la colonne nom.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | import csv nom_recherche = "Sara" resultat = None with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.DictReader(fichier) for ligne in lecteur: if ligne["nom"] == nom_recherche: resultat = ligne break if resultat: print("Étudiant trouvé :", resultat) else: print("Étudiant introuvable.") |
Sortie :
1 | Étudiant trouvé : {'id': '2', 'nom': 'Sara', 'age': '20', 'ville': 'Rabat', 'note': '17', 'filiere': 'Informatique'} |
Cette approche est généralement plus lisible, car le programme indique explicitement la colonne dans laquelle la recherche doit être effectuée.
Nous pouvons également rechercher un identifiant et afficher uniquement certaines informations de la ligne trouvée.
1 2 3 4 5 6 7 8 9 10 11 12 13 | import csv id_recherche = "5" with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.DictReader(fichier) for ligne in lecteur: if ligne["id"] == id_recherche: print("Nom :", ligne["nom"]) print("Ville :", ligne["ville"]) print("Note :", ligne["note"]) break |
Sortie :
1 2 3 | Nom : Omar Ville : Oujda Note : 14 |
Pour rechercher un enregistrement à partir d'un identifiant unique, l'utilisation de break est logique puisque nous n'attendons normalement qu'un seul résultat.
5. Rechercher plusieurs occurrences d'une même valeur
Une valeur peut apparaître plusieurs fois dans une même colonne. Par exemple, plusieurs étudiants peuvent habiter dans la même ville. Dans ce cas, il ne faut pas arrêter la boucle avec break après la première correspondance.
Le programme suivant recherche tous les étudiants habitant à Oujda.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 | import csv ville_recherchee = "Oujda" nombre = 0 with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.DictReader(fichier) for ligne in lecteur: if ligne["ville"] == ville_recherchee: print(ligne["nom"], "-", ligne["ville"]) nombre += 1 print("Nombre de résultats :", nombre) |
Sortie :
1 2 3 | Amine - Oujda Omar - Oujda Nombre de résultats : 2 |
Une autre possibilité consiste à stocker toutes les lignes correspondantes dans une liste. Cette méthode est pratique lorsque les résultats doivent être réutilisés après la fermeture du fichier.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | import csv ville_recherchee = "Rabat" resultats = [] with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.DictReader(fichier) for ligne in lecteur: if ligne["ville"] == ville_recherchee: resultats.append(ligne) print("Nombre de résultats :", len(resultats)) for etudiant in resultats: print(etudiant["nom"], "-", etudiant["note"]) |
Sortie :
1 2 3 | Nombre de résultats : 2 Sara - 17 Nadia - 16 |
Nous pouvons également utiliser une compréhension de liste lorsque nous souhaitons obtenir directement l'ensemble des résultats.
1 2 3 4 5 6 7 8 9 10 11 12 13 | import csv with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.DictReader(fichier) informaticiens = [ ligne for ligne in lecteur if ligne["filiere"] == "Informatique" ] for etudiant in informaticiens: print(etudiant["nom"]) |
Sortie :
1 2 3 | Sara Lina Nadia |
6. Effectuer une recherche partielle et ignorer la casse
Une comparaison avec l'opérateur == exige une correspondance exacte. Dans certaines applications, nous souhaitons au contraire effectuer une recherche partielle. L'opérateur in peut alors être utilisé sur les chaînes de caractères.
Le programme suivant recherche les noms contenant la chaîne "ar".
1 2 3 4 5 6 7 8 9 10 | import csv texte_recherche = "ar" with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.DictReader(fichier) for ligne in lecteur: if texte_recherche in ligne["nom"]: print(ligne["nom"]) |
Sortie :
1 2 | Sara Omar |
Une recherche peut également échouer uniquement à cause des majuscules et des minuscules. Pour rendre la comparaison insensible à la casse, nous pouvons utiliser la méthode casefold().
1 2 3 4 5 6 7 8 9 10 | import csv ville_recherchee = "oujda" with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.DictReader(fichier) for ligne in lecteur: if ligne["ville"].casefold() == ville_recherchee.casefold(): print(ligne["nom"], "-", ligne["ville"]) |
Sortie :
1 2 | Amine - Oujda Omar - Oujda |
Pour une recherche partielle insensible à la casse, nous pouvons combiner in et casefold().
1 2 3 4 5 6 7 8 9 10 | import csv texte_recherche = "INFO" with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.DictReader(fichier) for ligne in lecteur: if texte_recherche.casefold() in ligne["filiere"].casefold(): print(ligne["nom"], "-", ligne["filiere"]) |
Sortie :
1 2 3 | Sara - Informatique Lina - Informatique Nadia - Informatique |
La méthode casefold() est particulièrement adaptée aux comparaisons textuelles sans distinction entre majuscules et minuscules.
7. Rechercher avec plusieurs critères
Une recherche peut porter simultanément sur plusieurs colonnes. Les opérateurs logiques and et or permettent de combiner plusieurs conditions.
Le programme suivant recherche les étudiants habitant à Rabat et appartenant à la filière Informatique.
1 2 3 4 5 6 7 8 9 10 11 | import csv with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.DictReader(fichier) for ligne in lecteur: if ( ligne["ville"] == "Rabat" and ligne["filiere"] == "Informatique" ): print(ligne["nom"], "-", ligne["note"]) |
Sortie :
1 2 | Sara - 17 Nadia - 16 |
L'opérateur or permet au contraire de retenir une ligne lorsque l'une au moins des conditions est satisfaite. L'exemple suivant recherche les étudiants habitant Oujda ou Fès.
1 2 3 4 5 6 7 8 | import csv with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.DictReader(fichier) for ligne in lecteur: if ligne["ville"] == "Oujda" or ligne["ville"] == "Fès": print(ligne["nom"], "-", ligne["ville"]) |
Sortie :
1 2 3 | Amine - Oujda Yassine - Fès Omar - Oujda |
Pour plusieurs valeurs possibles dans une même colonne, l'opérateur in associé à une collection rend souvent le test plus compact.
1 2 3 4 5 6 7 8 9 10 | import csv villes = {"Oujda", "Rabat"} with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.DictReader(fichier) for ligne in lecteur: if ligne["ville"] in villes: print(ligne["nom"], "-", ligne["ville"]) |
Sortie :
1 2 3 4 | Amine - Oujda Sara - Rabat Omar - Oujda Nadia - Rabat |
8. Rechercher des valeurs numériques dans un fichier CSV
Les données récupérées avec csv.DictReader() sont généralement des chaînes de caractères. Pour effectuer une recherche numérique basée sur une comparaison comme >, <, >= ou <=, il faut convertir la valeur avec int() ou float().
Le programme suivant recherche les étudiants dont la note est supérieure ou égale à 16.
1 2 3 4 5 6 7 8 9 10 11 12 | import csv note_minimale = 16 with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.DictReader(fichier) for ligne in lecteur: note = float(ligne["note"]) if note >= note_minimale: print(ligne["nom"], "-", note) |
Sortie :
1 2 3 | Sara - 17.0 Lina - 18.0 Nadia - 16.0 |
Nous pouvons également rechercher les valeurs appartenant à un intervalle numérique. L'exemple suivant sélectionne les étudiants dont la note est comprise entre 14 et 17 inclus.
1 2 3 4 5 6 7 8 9 10 | import csv with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.DictReader(fichier) for ligne in lecteur: note = float(ligne["note"]) if 14 <= note <= 17: print(ligne["nom"], "-", note) |
Sortie :
1 2 3 4 | Amine - 15.5 Sara - 17.0 Omar - 14.0 Nadia - 16.0 |
La même technique peut être appliquée aux âges. Nous convertissons alors la valeur avec int().
1 2 3 4 5 6 7 8 9 10 | import csv age_recherche = 21 with open("etudiants.csv", "r", encoding="utf-8", newline="") as fichier: lecteur = csv.DictReader(fichier) for ligne in lecteur: if int(ligne["age"]) == age_recherche: print(ligne["nom"], "-", ligne["age"], "ans") |
Sortie :
1 2 | Amine - 21 ans Omar - 21 ans |
Il est important d'effectuer cette conversion avant les comparaisons numériques afin d'éviter de comparer les nombres comme de simples chaînes de caractères.
9. Créer une fonction réutilisable de recherche dans un fichier CSV
Lorsque des recherches doivent être effectuées régulièrement, il est préférable de créer une fonction réutilisable. La fonction suivante reçoit le nom du fichier, la colonne à examiner et la valeur recherchée.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | import csv def rechercher_csv(nom_fichier, colonne, valeur): resultats = [] with open( nom_fichier, "r", encoding="utf-8", newline="" ) as fichier: lecteur = csv.DictReader(fichier) for ligne in lecteur: if ligne[colonne] == valeur: resultats.append(ligne) return resultats resultats = rechercher_csv( "etudiants.csv", "ville", "Oujda" ) for etudiant in resultats: print(etudiant["nom"], "-", etudiant["ville"]) |
Sortie :
1 2 | Amine - Oujda Omar - Oujda |
L'intérêt de cette fonction est qu'elle peut rechercher une valeur dans différentes colonnes sans modifier son code interne.
1 2 3 4 5 6 7 8 9 10 | resultats = rechercher_csv( "etudiants.csv", "filiere", "Informatique" ) print("Nombre de résultats :", len(resultats)) for etudiant in resultats: print(etudiant["nom"]) |
Sortie :
1 2 3 4 | Nombre de résultats : 3 Sara Lina Nadia |
Nous pouvons améliorer la fonction afin d'effectuer une recherche textuelle insensible à la casse. Il est également prudent de vérifier que la colonne demandée existe dans le fichier.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | import csv def rechercher_texte(nom_fichier, colonne, valeur): resultats = [] with open( nom_fichier, "r", encoding="utf-8", newline="" ) as fichier: lecteur = csv.DictReader(fichier) if colonne not in lecteur.fieldnames: return resultats for ligne in lecteur: if ligne[colonne].casefold() == valeur.casefold(): resultats.append(ligne) return resultats resultats = rechercher_texte( "etudiants.csv", "ville", "oujda" ) for ligne in resultats: print(ligne["nom"]) |
Sortie :
1 2 | Amine Omar |
Une fonction de ce type peut servir de base à un programme de consultation de données, à un formulaire de recherche ou à une petite application utilisant des fichiers CSV.
10. À retenir !
La recherche d'une valeur dans un fichier CSV en Python consiste principalement à parcourir les lignes du fichier puis à comparer une ou plusieurs colonnes avec les critères recherchés.
- csv.reader() permet de rechercher une valeur en travaillant avec les indices des colonnes.
- csv.DictReader() permet de rechercher directement à partir des noms des colonnes.
- L'opérateur == permet d'effectuer une recherche exacte.
- L'opérateur in permet notamment d'effectuer une recherche partielle dans une chaîne.
- break permet d'arrêter la recherche après la première correspondance.
- Pour obtenir toutes les correspondances, il faut continuer à parcourir le fichier et éventuellement stocker les résultats dans une liste.
- La méthode casefold() permet d'effectuer des recherches textuelles sans tenir compte de la casse.
- Les opérateurs and et or permettent de combiner plusieurs critères.
- Les valeurs numériques doivent généralement être converties avec int() ou float() avant une comparaison numérique.
- Une fonction de recherche permet de rendre le programme plus générique et réutilisable.
Le modèle suivant résume une méthode simple permettant de rechercher toutes les lignes correspondant à une valeur dans une colonne donnée.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | import csv def rechercher_csv(nom_fichier, colonne, valeur): resultats = [] with open( nom_fichier, "r", encoding="utf-8", newline="" ) as fichier: lecteur = csv.DictReader(fichier) for ligne in lecteur: if ligne[colonne] == valeur: resultats.append(ligne) return resultats resultats = rechercher_csv( "etudiants.csv", "ville", "Rabat" ) for ligne in resultats: print(ligne["nom"], "-", ligne["ville"]) |
Sortie :
1 2 | Sara - Rabat Nadia - Rabat |
Auteur : Younes Derfoufi
Lieu de travail : CRMEF OUJDA
Site Web : www.tresfacile.net
Chaine YouTube : https://www.youtube.com/user/InformatiquesFacile
Me contacter : https://www.tresfacile.net/me-contacter/
