python - 在指定行 Python 之后拆分 CSV 文件

Question

我正在尝试拆分一个包含 201 行的 CSV 文件。前 136 行包含与其余 65 行不同的信息。无论如何，我可以在 136 行之后拆分 CSV 文件吗？

到目前为止我找到的解决方案平均分割文件。我也将此作为可能的解决方案进行了研究:https://gist.github.com/jrivero/1085501

Answer 1

您可以使用两个 csv.reader文件对象上的对象。使用 itertools.islice 将第一个读取器对象切片到指定的行数, 然后使用第二个阅读器阅读其余行。

如果这两个部分的分隔符不同，两个 csv.reader 对象可以轻松处理:

import csv
from itertools import islice

with open('your_file_name.csv') as f:
    first_rows = list(islice(csv.reader(f), None, 136))
    last_rows = list(csv.reader(f, delimiter=';')) # use a different delim or quote char