在指定的行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

Answer 2

如果您知道需要跳過的行數，則可以使用csvreader.line_num來了解正在閱讀的行，您可以在此處找到有關此行的更多信息： https : csvreader.line_num .html＃reader-objects