Sauter les premières lignes lors de la lecture des lignes d'un fichier Python

Question

Je veux sauter les 17 premières lignes lors de la lecture d'un fichier texte.

Disons que le fichier ressemble à :

0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
0
good stuff

Je veux juste les bonnes choses. Ce que je fais est beaucoup plus compliqué, mais c'est la partie qui me pose problème.

Answer 1

Utilisez une tranche, comme ci-dessous :

with open('yourfile.txt') as f:
    lines_after_17 = f.readlines()[17:]

---

Si le fichier est trop volumineux pour être chargé en mémoire :

with open('yourfile.txt') as f:
    for _ in range(17):
        next(f)
    for line in f:
        # do stuff

Answer 2

Utilisez itertools.islice à partir de l'indice 17. Il sautera automatiquement les 17 premières lignes.

import itertools
with open('file.txt') as f:
    for line in itertools.islice(f, 17, None):  # start=17, stop=None
        # process lines

Answer 3

for line in dropwhile(isBadLine, lines):
    # process as you see fit

Démonstration complète :

from itertools import *

def isBadLine(line):
    return line=='0'

with open(...) as f:
    for line in dropwhile(isBadLine, f):
        # process as you see fit

Avantages : Ceci est facilement extensible aux cas où vos lignes de préfixe sont plus compliquées que "0" (mais pas interdépendantes).

Answer 4

Utilisez f.readlines(17) . Après cela, si vous itérez sur l'objet fichier ou si vous utilisez la fonction f.readline() vous serez à la ligne 18.

Answer 5

Voici les résultats de timeit pour les 2 premières réponses. Notez que "file.txt" est un fichier texte contenant plus de 100 000 lignes de chaînes de caractères aléatoires avec une taille de fichier de plus de 1 Mo.

Utilisation de itertools :

import itertools
from timeit import timeit

timeit("""with open("file.txt", "r") as fo:
    for line in itertools.islice(fo, 90000, None):
        line.strip()""", number=100)

>>> 1.604976346003241

En utilisant deux boucles for :

from timeit import timeit

timeit("""with open("file.txt", "r") as fo:
    for i in range(90000):
        next(fo)
    for j in fo:
        j.strip()""", number=100)

>>> 2.427317383000627

Il est clair que la méthode itertools est plus efficace lorsqu'il s'agit de gros fichiers.