python - 如何删除以 "-"开头的行以及之后的 3 行

Question

我有这样的测试数据。它包含 1 列 .CSV，其中包含

"----------"

PAGE1

PARAGRAPH

EXAMPLE

example1

example2

example3

example4

example5

"----------"

PAGE2

PARAGRAPH

EXAMPLE

example1

example2

example3

example4

example5

目的是删除包含

的 4 行

"------"

PAGE

PARAGRAPH

EXAMPLE

这样我就可以输出一个示例列表

import csv
input = open('Test_Parse.csv', 'rb')
output = open('first_edit.csv', 'wb')
writer = csv.writer(output)
for row in csv.reader ('Test_Parse.csv'):
if not row.startswith ("------"):
    writer.writerow(row)
input.close()
output.close()

这是尝试删除带有 "-----" 的行，但很困难？

任何帮助或指出正确的方向将不胜感激!

Answer 1

将fileobject视为迭代器:

import csv

with open('Test_Parse.csv', 'r') as inp, open('first_edit.csv', 'w', newline='') as out:
    writer = csv.writer(out)
    for l in inp:
        if l.startswith('"------'):
            next(inp)  # extract the next line from the file to skip
            next(inp)
            next(inp)
        else:
            writer.writerow((l.strip(),))

<小时/>

最终的first_edit.csv内容:

example1
example2
example3
example4
example5
example1
example2
example3
example4
example5