如何开始写入文本文件2行

Question

我当前正在打开一个csv文件，并将其写入新的.h文件。但是，我想将所有内容都写到csv文件的前两行。

如何开始写两行？

import csv
newheader2 = open("ffile2.h", 'w')
with open("pcf_gras.csv") as newcsvfile:
    readCSV = csv.reader(newcsvfile, delimiter= ',')
    newheader2.write("Heading Comment Section")
    for item in readCSV:
        #name = item[2]
        #newheader2.write(name)
        #print(item[2], item[5])
        newheader2.write("#define ")
        newheader2.write(item[2])
        newheader2.write(" ")
        newheader2.write(item[5])
        newheader2.write("\n")

Answer 1

创建readCSV对象后，可以通过调用以下内容跳过两行：

next(readCSV)
next(readCSV)

然后，当您执行for item in readCSV:时，它将从文件中的该位置开始。