Python从文件中提取数据

Question

我正在尝试提取具有特定文本文件的文本:

----
data1
data1
data1
extractme
----
data2
data2
data2
----
data3
data3
extractme
----

然后转存到文本文件中

----
data1
data1
data1
extractme
---
data3
data3
extractme
---

感谢您的帮助。

Answer 1

这对我来说效果很好。您的示例数据位于名为“data.txt”的文件中，输出转到“result.txt”

inFile = open("data.txt")
outFile = open("result.txt", "w")
buffer = []
keepCurrentSet = True
for line in inFile:
    buffer.append(line)
    if line.startswith("----"):
        #---- starts a new data set
        if keepCurrentSet:
            outFile.write("".join(buffer))
        #now reset our state
        keepCurrentSet = False
        buffer = []
    elif line.startswith("extractme"):
        keepCurrentSet = True
inFile.close()
outFile.close()