Python Grep Wort und Zeile vor diesem Wort

Question

Ich sollte eine Zeile finden. Ich habe ein Skript geschrieben. Ich habe eine Datei, wo ich Zeile duke

finden sollte

#!/usr/bin/python3.5 

for line in open('test.txt'): 
    if 'duke' in line: 
     print(line) 

Alles ist in Ordnung. Aber wie kann ich 2 Zeilen vor diesem Ergebnis sehen?

In Grep war es grep -B 2 "duke" test.txt. Wie kann ich das gleiche in Python tun?

Answer 1

Sie benötigen einen Puffer mit N vorhergehenden Zeilen zu halten (in diesem Fall N = 2). Beispiel:

N=2 
buffer = [] 
for line in open('test.txt'): 
    buffer.append(line) 
    if len(buffer) > N: 
     buffer.pop(0) 
    if 'duke' in line: 
     print(buffer) 

Answer 2

Sie können ein deque mit einer festen Größe von 2 verwenden:

#!/usr/bin/python3.5 
from collections import deque 

queue = deque(maxlen=2) 

for line in open('test.txt'): 
    if 'duke' in line: 
     print(list(queue)) 
    queue.append(line)