python 将字典写入csv

import os
import json
import csv
import os.path
from pathlib import Path
# os.path.isfile(fname)

root='/faster_rcnn'
save='sum.csv'

h=['name','AP','AP50','AP75','bAP','bAP50','bAP75','nAP','nAP50','nAP75']

with open(save,'w',newline='') as f:
    writer = csv.DictWriter(f,fieldnames=h)
    writer.writeheader()
    for d in os.listdir(root):
        tmp_root = root + '/' + d
        for dd in os.listdir(tmp_root):
            if dd == 'inference':
                cur = tmp_root + '/' + dd + '/res_final.json'
                if  Path(cur).exists():
                    with open(cur) as fp:
                        read = json.load(fp)
                        dic = read['bbox']
                        dic['name'] = d
                        print(d)
                        writer.writerow(dic)

使用csv包
主要语句:

h=['name','AP','AP50','AP75','bAP','bAP50','bAP75','nAP','nAP50','nAP75']
with open(save,'w',newline='') as f:
    writer = csv.DictWriter(f,fieldnames=h)   #构造一个write
    writer.writeheader()    #写入表头
    writer.writerow(dic)    #将数据写入csv,会根据字典的键值对自动匹配表头

如果表头是字典形式,写入顺序和字典定义顺序不同,如果表头h定义的时候是lsit形式,在csv文件中写入的表头的顺序和定义的h顺序一样

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值