最近要利用模型检测的结果做数据集,所以要把检测到的boundingbox等信息写入csv文件。看了下别人的,主要有两种方式
一、采用csv模块
参考博客:https://blog.csdn.net/lbj1260200629/article/details/89600055(本人借用的方式)
# 导入CSV安装包
import csv
# 1. 创建文件对象
f = open('文件名.csv','w',encoding='utf-8')
# 2. 基于文件对象构建 csv写入对象
csv_writer = csv.writer(f)
# 3. 构建列表头
csv_writer.writerow(["姓名","年龄","性别"])
# 4. 写入csv文件内容
csv_writer.writerow(["l",'18','男'])
csv_writer.writerow(["c",'20','男'])
csv_writer.writerow(["w",'22','女'])
# 5. 关闭文件
f.close()
二、利用pandas第三方库
参考博客:https://www.cnblogs.com/beyondChan/p/10776345.html