Comment extraire les N premières lignes d’un fichier en Python ?

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

Souvent, lorsque l'on travaille avec de gros fichiers de données brutes, 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 :

<code class="python">with open(path_to_file) as input_file:
    head = [next(input_file) for _ in range(lines_number)]
print(head)</code>

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 :

<code class="python">from itertools import islice

with open(path_to_file) as input_file:
    head = list(islice(input_file, lines_number))
print(head)</code>

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

Effet du 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.

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!