晚上在跑代码的时候 想把结果通过保存成csv文件的形式 遇到了这个问题
dit = {'result_name': [image_path], 'psnr': psnr_list, 'ssim': ssim_list}
df = pd.DataFrame(dit)
csv_path = 'csv/'+'ssim&psnr.csv' # 拼接csv名字# csv全路径
df.to_csv(csv_path, columns=['result_name', 'psnr', 'ssim'], index=False, sep=',')
先看改后的代码
问题出在了pd.DataFrame(dit)和dit={}这两行中
①如果dit中存储的都是单个数据 则考虑在
pd.DataFrame(dit)中改成 pd.DataFrame(dit,index[0])
②如果dit中 像上述描述的代码中 将单个数据用列表存储
tips:代码中的psnr_list和ssim_list都是以列表的形式存储的哦~