python - 使用 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')