Récupération des N premières lignes d'un fichier

Souvent, lorsque vous travaillez avec des fichiers de données brutes volumineux, il devient nécessaire d'extraire un nombre spécifique de lignes pour un traitement ou une analyse ultérieurs. En Python, il existe plusieurs approches pour accomplir cette tâche.

Lecture des N premières lignes à l'aide de la compréhension de liste

Une méthode simple et efficace consiste à utiliser la compréhension de liste :

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

Cette approche parcourt le fichier d'entrée à l'aide de la fonction next() et stocke les premières lignes lines_number dans la liste d'en-tête.

Utilisation de la fonction islice()

Une autre approche exploite le module itertools de Python :

from itertools import islice

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

Ici, la fonction islice() est utilisée pour parcourir les premières lignes lines_number du fichier d'entrée, créant une liste des lignes extraites.

Effet de Système d'exploitation

L'implémentation décrite ci-dessus devrait fonctionner quel que soit le système d'exploitation utilisé. Cependant, il convient de noter que dans Python 2, la fonction next() est connue sous le nom de xrange(), ce qui peut nécessiter des ajustements correspondants dans les anciennes bases de code.