Jak odczytywać wcześniej plik podczas przechodzenia przez niego za pomocą Pythona?

Question

Pętlę przechodzę przez plik i jeśli coś znajdę, chciałbym przeczytać przed sobą kilka linii, szukając czegoś, zanim wróci kontrola do głównej pętli. Jednak chcę zwrócić kontrolę w punkcie, w którym przestałem patrzeć w przyszłość.

Przykładowy kod:

for line in file: 
    line = line.strip() 
    llist = line.split() 

    if llist[0] == 'NUMS': 
     # some loop to read ahead and print nums on their own line 
     # until it finds END, then return control to the main for 
     # loop at the point where it stopped looking ahead. 

Wejście próbki:

NUMS 
1 
2 
3 
4 
5 
END 
SOME 
MORE 
STUFF 
NUMS 
6 
7 
8 
9 
0 
END 

Pożądany wyjściowa:

1 2 3 4 5 
6 7 8 9 0 

Jestem dość nowy w Pythonie, więc jeśli istnieje lepszy sposób to zrobić oprócz używania pętli do patrzenia w przyszłość, cieszę się, że to widzę.

Answer 1

Po zapętleniu otwartych plików, można uzyskać tylko jedną linię dokładnie jeden raz. W przeciwieństwie do list, gdzie każda pętla for otrzyma własny iterator na całej liście.

Można mieć wewnętrzną pętlę „ukraść” wiersze z pliku, tak, że zewnętrzna pętla nie dostać je zobaczyć:

for line in file: 
    line = line.strip() 
    llist = line.split() 

    if llist and llist[0] == 'NUMS': 
     for line in file: 
      line = line.strip() 
      if line == 'END': 
       break 
      else: 
       print line, 
     print # newline 

Poczytaj na ten temat, jak iterators praca uzyskać więcej informacji.

Answer 2

Nie warto czytać dalej, jeśli nie jest to konieczne, co ma miejsce w tym przypadku. To, co musisz zrobić, można osiągnąć w pojedynczej pętli for, utrzymując jeden bit stanu. To pozwala na sprawdzanie błędów i skaluje się do znacznie bardziej skomplikowanych wymagań niż może być obsługiwane przez zagnieżdżone pętle nad tym samym iteratorem.

guff = """NUMS 
1 
2 
etc etc 
9 
0 
END""" 

state = "" 
for item in guff.split(): 
    if item == "NUMS": 
     if state == "NUMS": print "missing END" 
     state = "NUMS" 
    elif item == "END": 
     if state == "": print "missing NUMS" 
     state = "" 
     print 
    elif state == "NUMS": 
     print item, 

if state == "NUMS": print # missing END