使用python格式化/插入数据行中的逗号

Question

我有一个定期附加的CSV文件，其中包含两种类似的数据。我无法控制数据如何进入CSV - 但我想执行一个python脚本来格式化CSV，以便我可以将其摄取到数据库中。这就是我想要做的事情：

传入的原始数据如下所示：

record, date & time, sensor, height, status

record, date & time, sensor, height, weight, status

如果没有重量或类似的东西，我们的想法就是在高度之后迭代地添加这些数据。基本上模仿一个空白字段，以便数据基本上如下所示：

record, date & time, sensor, height, **weight,**status (updated after python script if/then)

record, date & time, sensor, height, weight, status

希望这是有意义的 - 基本上添加一个字段，我会在每次更新文件时运行脚本。

Answer 1

import csv
import os
with open('yourfile.csv') as f_in, open('yourfile.csv.temp') as f_out:
    reader = csv.reader(f_in)
    writer = csv.writer(f_out)
    for row in reader:
        if len(row) == 5:
            row.insert(4, '')
        writer.writerow(row)
os.rename('yourfile.csv.temp', 'yourfile.csv')