python - 如何以 csv 格式正确添加我的数据？

Question

我将数据写入 .txt 文件的代码:

with open(file_path, 'w') as f:
    id = 1
    for line in value:
        line = re.sub('[^A-Za-z0-9-,]+', '', str(line))
        ymax, xmax, xmin, ymin=line.split(',')
        f.write(('{{\'yMax\': u\'{}\', \'xMax\': u\'{}\', \'xMin\': u\'{}\',\'yMin\': u\'{}\', \'id\': \'{}\', \'name\': \'\'}}'.format(ymax, xmax, xmin, ymin,id)))
        id = id + 1

结果:

{'yMax': u'156', 'xMax': u'4802', 'xMin': u'4770','yMin': u'141', 'id': '1', 'name': ''}
{'yMax': u'157', 'xMax': u'4895', 'xMin': u'4810','yMin': u'141', 'id': '2', 'name': ''}

但是我希望我的数据以类似 .csv 格式的表格形式存在:

image     id    name    xMin    xMax    yMin    yMax
1-0.png   1             4770    4802    141     156
1-0.png   2             4810    4895    141     157

我如何调整我的代码以从我已有的 .txt 格式转换为我想要的 .csv 格式？图像的多余列只是 txt 的文件名，但 .png 而不是 txt 使用 re 很容易我可以稍后调整我的主要问题是表格形状。

Answer 1

这应该有所帮助。使用 csv.DictWriter

with open(file_path, 'w') as f:
    writer = csv.DictWriter(f, delimiter='\t', fieldnames=['yMax', 'xMax', 'xMin', 'yMin', 'id', 'name'])      #Tab seperated
    writer.writeheader()    #Add header
    for i, line in enumerate(value, 1):             #enumerate to get id
        line = re.sub('[^A-Za-z0-9-,]+', '', str(line))
        ymax, xmax, xmin, ymin=line.split(',')
        d = {'yMax': ymax,'xMax': xmax, 'xMin': xmin,'yMin': ymin, 'id': i, 'name': ''}
        writer.writerow(d)

注意:这是示例代码。