在开始和停止标志之间读取多个文件块

Question

我正在尝试将文件的各个部分读取为numpy数组，这些数组对于文件的不同部分具有相似的开始和停止标志。目前，我已经找到了一种有效的方法，但是仅在输入文件的一部分中需要重新打开输入文件。

目前，我的代码是：

    with open("myFile.txt") as f:
        array = []
        parsing = False
        for line in f:
            if line.startswith('stop flag'):
            parsing = False
        if parsing:
            #do things to the data
        if line.startswith('start flag'):
            parsing = True

我从这个问题中找到了代码

使用此代码，我需要重新打开并阅读文件。

有没有一种方法可以读取所有部分，而不必打开要读取的每个部分的文件？

Answer 1

每次到达开始标记直到停止都可以使用itertools.take：

from itertools import takewhile
with open("myFile.txt") as f:
        array = []
        for line in f:
            if line.startswith('start flag'):               
                data = takewhile(lambda x: not x.startswith("stop flag"),f)
                # use data and repeat

或者只是使用一个内循环：

with open("myFile.txt") as f:
    array = []
    for line in f:
        if line.startswith('start flag'):
            # beginning of section use first lin
            for line in f:
                # check for end of section breaking if we find the stop lone
                if line.startswith("stop flag"):
                    break
                 # else process lines from section

文件对象返回其自己的迭代器，因此当您遍历时f，指针将继续移动，当您到达start标志时，开始处理一段，直到您到达停靠点为止。根本没有理由重新打开文件，只需在文件的各行中迭代一次即可使用这些部分。如果开始和停止标志线被认为是该部分的一部分，请确保也使用它们。