Lecture des N premières lignes d'un fichier en Python

Dans le traitement des données, il est souvent nécessaire de manipuler seulement une partie d'un grand ensemble de données . C'est là qu'intervient la possibilité de lire un nombre spécifié de lignes à partir d'un fichier texte.

La méthode intégrée de Python

Python fournit un moyen simple de lire les N premières lignes d'un fichier :

with open(path_to_file) as input_file:
    head = [next(input_file) for _ in range(lines_number)]

Cet extrait de code ouvre le fichier au chemin spécifié, puis parcourt les lignes lines_number fois, stockant les lignes récupérées dans la liste principale.

Considérations relatives au système d'exploitation

Le système d'exploitation n'affecte généralement pas la mise en œuvre de cette tâche. Python gère l'accès aux fichiers quel que soit le système sous-jacent, ce qui rend le code portable sur différentes plates-formes.

Approche alternative utilisant islice

Une autre option pour lire les N premières lignes consiste à utilisez la fonction islice du module itertools :

from itertools import islice

with open(path_to_file) as input_file:
    head = list(islice(input_file, lines_number))

Cette méthode renvoie un itérateur générateur qui produit les premières lignes lines_number du fichier. La fonction list() est utilisée pour convertir le générateur en liste pour une manipulation facile.

Conclusion

Ces extraits de code fournissent des moyens fiables de lire un nombre spécifié de lignes à partir d'un fichier texte en Python. Que vous souhaitiez réduire un grand ensemble de données ou effectuer une opération spécifique sur les premières lignes, ces méthodes offrent des solutions efficaces et polyvalentes.