python读写csv文件

import csv
import os
import math

# 1、读取csv数据
with open('./data/image.csv',encoding='utf-8')as fp:
    reader = csv.DictReader(fp)
    datas=[]
    for i in reader:
        if '.jpg' in i['fy']:
            data={}
            data["name"]=os.path.basename(i['fy'])[:-4]
            data["x"]=i['cx']
            data["y"]=i['cy']
            data["z"]=i['width']
            data["rot_z"]=float(i['offset_y'])+math.pi/2
            datas.append(data)
header = ['name','x','y','z','rot_z']

# 2、写入csv数据  
with open ('./data/camerainfo.csv','w',encoding='utf-8',newline='') as fp:
    writer =csv.DictWriter(fp,header)
    # 写入标题
    writer.writeheader()
    # 将数据写入
    writer.writerows(datas)
print("ok")