Comment lire les N premières lignes d'un fichier ?

Question

Nous disposons d'un grand fichier de données brutes que nous souhaitons réduire à une taille donnée.

Comment obtenir les N premières lignes d'un fichier texte en python ? Est-ce que le système d'exploitation utilisé aura un effet sur la mise en oeuvre ?

Answer 1

Python 3 :

with open("datafile") as myfile:
    head = [next(myfile) for x in range(N)]
print(head)

Python 2 :

with open("datafile") as myfile:
    head = [next(myfile) for x in xrange(N)]
print head

Voici une autre façon de procéder (à la fois pour Python 2 et 3) :

from itertools import islice

with open("datafile") as myfile:
    head = list(islice(myfile, N))
print(head)

Answer 2

N = 10
with open("file.txt", "a") as file:  # the a opens it in append mode
    for i in range(N):
        line = next(file).strip()
        print(line)

Answer 3

Si vous voulez lire les premières lignes rapidement et que vous ne vous souciez pas de la performance, vous pouvez utiliser .readlines() qui renvoie un objet liste, puis découpe la liste.

Par exemple, pour les 5 premières lignes :

with open("pathofmyfileandfileandname") as myfile:
    firstNlines=myfile.readlines()[0:5] #put here the interval you want

Nota: le fichier entier est lu, il en est de même pour pas le meilleur du point de vue de la performance b est facile à utiliser, rapide à écrire et facile à mémoriser. quelques calculs ponctuels, c'est très pratique

print firstNlines

Un avantage par rapport aux autres réponses est la possibilité de sélectionner facilement la plage de lignes, par exemple en sautant les 10 premières lignes. [10:30] ou les 10 dernières [:-10] ou en ne prenant que les lignes paires [::2] .

Answer 4

Ce que je fais, c'est appeler les lignes N en utilisant pandas . Je pense que la performance n'est pas la meilleure, mais par exemple si N=1000 :

import pandas as pd
yourfile = pd.read_csv('path/to/your/file.csv',nrows=1000)

Answer 5

Il n'existe pas de méthode spécifique pour lire le nombre de lignes indiqué par l'objet fichier.

Je pense que la méthode la plus simple serait la suivante :

lines =[]
with open(file_name) as f:
    lines.extend(f.readline() for i in xrange(N))