混合文件和循环

Question

我正在编写一个脚本来记录来自另一个程序的错误,并在遇到错误时重新启动它停止的程序.无论出于何种原因,该程序的开发人员都认为没有必要在默认情况下将此功能放入其程序中.

无论如何,程序接受一个输入文件,解析它,并创建一个输出文件.输入文件采用特定格式:

UI - 26474845
TI - the title (can be any number of lines)
AB - the abstract (can also be any number of lines)

当程序抛出错误时,它会为您提供跟踪错误所需的参考信息 - 即UI,哪个部分(标题或摘要)以及相对于标题或摘要开头的行号.我想从输入文件中记录有问题的句子,其中包含一个带参考号和文件的函数,找到句子并记录它.我能想到的最好的方法是在文件中向前移动特定次数(即n次,其中n是相对于seciton开头的行号).这样做的方式似乎是:

i = 1
while i <= lineNumber:
    print original.readline()
    i += 1

我不知道这会如何让我丢失数据,但Python认为它会,并说ValueError: Mixing iteration and read methods would lose data.有谁知道如何正确地做到这一点？

Answer 1

你得到了ValueError异常,因为你的代码可能有for line in original:除了original.readline().一个简单的解决方案,可以解决问题,而不会使您的程序更慢或消耗更多的内存正在改变

for line in original:
    ...

至

while True:
    line = original.readline()
    if not line: break
    ...

Answer 2

使用for和枚举.

例:

for line_num, line in enumerate(file):
    if line_num < cut_off:
        print line

注意:这假设您已经清理了文件句柄等.

此外,如果您喜欢更实用的味道,则takewhile函数可能会很有用.