Python grep單詞並在此單詞之前顯示行

Question

我應該找到一些行。我寫了劇本。我有一個文件，我應該在哪裏找到線duke

#!/usr/bin/python3.5 

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

一切正常。但是如何在這個結果之前看到2行？

在grep中是grep -B 2 "duke" test.txt。我如何在Python中做同樣的事情？

Answer 1

你需要保持與N-先前行緩衝器（在這種情況下，N = 2）。例如：

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

可以使用deque爲2的固定大小：

#!/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)