如何将这些数据保存到csv文件中?

时间:2019-04-05 05:30:49

标签: python python-3.x csv export-to-csv

我有一条看起来像这样的数据。如何将其保存到csv文件中?我当时正在考虑将其添加到另一个列表中,但不确定是否可行。我尝试使用下面链接中提出的问题的解决方案,但是csv文件仅保存最后一张图片的坐标。

How do I write this piece of data to a csv file?

data = 
[{'box': [43, 37, 133, 168], 
 'confidence': 0.99, 
 'keypoints': {'left_eye': (78, 104), 
               'right_eye': (143, 99), 
               'nose': (110, 137), 
               'mouth_left': (82, 161), 
               'mouth_right': (147, 156)}}]

[{'box': [34, 33, 119, 161], 
  'confidence': 0.99,
  'keypoints': {'left_eye': (61, 104), 
                'right_eye': (116, 99), 
                'nose': (86, 133), 
                'mouth_left': (67, 161), 
                'mouth_right': (117, 156)}}]

这是我第一次问一个问题,如果不清楚,请原谅我!

我的预期输出是:sample

1 个答案:

答案 0 :(得分:0)

import csv

fields = ['box', 'confidence', 'left_eye', 'right_eye', 'nose', 'mouth_left', 'mouth_right']

with open('output.csv', 'w') as file:
    writer = csv.DictWriter(file, fieldnames=fields)
    writer.writeheader()
    for item in data:
        writer.writerow({
            'box': item['box'],
            'confidence': item['confidence'],
            'left_eye': item['keypoints']['left_eye'],
            'right_eye': item['keypoints']['right_eye'],
            'nose': item['keypoints']['nose'],
            'mouth_left': item['keypoints']['mouth_left'],
            'mouth_right': item['keypoints']['mouth_right'],
        })