如何从特定行之后 2 行的所有行中计算一个单词？

Question

所以，这听起来可能有点令人困惑，我会尝试解释一下。例如从这些行：

next line 1
^^^^^^^^^^^^^^^^^^
red blue dark ten lemon
next line 2
^^^^^^^^^^^^^^^^^^^
hat 45 no dad fate orange
next line 3
^^^^^^^^^^^^^^^^^^^
tan rat lovely lemon eat 
you him lemon Daniel her"

我只对上面有“下一行”两行的行中的“柠檬”计数感兴趣。所以，我期望的输出是“2个柠檬”。

任何帮助将不胜感激！

到目前为止我的尝试是：

#!/usr/bin/env python
#import the numpy library
 import numpy as np

  lemon = 0

  logfile = open('file','r')

  for line in logfile:

  words = line.split()

  words = np.array(words)
  if np.any(words == 'next line'):
    if np.any(words == 'lemon'):
        lemon +=1
print "Total number of lemons is %d" % (lemon)

但只有当它与“下一行”在同一行时才算“柠檬”。

Answer 1

对于每一行，您需要能够访问它之前的两行。为此，您可以使用itertools.tee 来创建两个独立的文件对象（它们是类似迭代器的对象），然后使用itertools.izip() 来创建您期望的对：

from itertools import tee, izip
with open('file') as logfile:
    spam, logfile = tee(logfile)
    # consume first two line of spam
    next(spam)
    next(spam)
    for pre, line in izip(logfile, spam):
        if 'next line' in pre:
             print line.count('lemon')

或者，如果您只想计算行数，您可以在 sum() 中使用生成器表达式：

from itertools import tee, izip
with open('file') as logfile:
    spam, logfile = tee(logfile)
    # consume first two lines of spam
    next(spam)
    next(spam)
    print sum(line.count('lemon') for pre, line in izip(logfile, spam) if 'next line' in pre)

Answer 2

您可以遍历文件（这是一个迭代器）并在找到next line 行时调用next 两次，然后count lemon 出现的频率，for 循环并且对next 的调用使用相同的迭代器。

with open("data.txt") as f:
    lemon_count = 0
    for line in f:
        if "next line" in line:
            next(f) # skip next line
            lemon_count += next(f).count("lemon") # get count for next-next line

对于您的示例，lemon_count 最终为 2。这是假设在next 行和lemon 行之间没有其他next 行，或者lemon 行本身就是next 行。