通过 Python 在 CSV 的第二行插入新行

Question

是否可以在 CSV 文件的第二行插入新行？例如，我有一个包含列名称及其值的 CSV：

meter_id, sdp_id, lat, lon
813711, 1331, 34.298, -83.113

我尝试读取此文件的应用程序需要添加一个新行来指示列类型。在上面的示例中，所有内容都是字符串，因此 CSV 需要是：

meter_id, sdp_id, lat, lon
String, String, String, String
813711, 1331, 34.298, -83.113

我读过几篇如何在 CSV 末尾添加新行的文章，但找不到有关如何执行上述操作的任何内容。

Answer 1

这是使用csv模块的一种方法。

演示：

import csv
toAdd = ["String", "String", "String", "String"]
with open(filename, "r") as infile:
    reader = list(csv.reader(infile))
    reader.insert(1, toAdd)

with open(filename, "w") as outfile:
    writer = csv.writer(outfile)
    for line in reader:
        writer.writerow(line)