plt.savefig保存不了图片,程序没反应,卡在这条命令上

# -*- coding:utf-8
import matplotlib as mpl
mpl.use('Agg')
import csv
import matplotlib as mpl
import matplotlib.pyplot as plt

#输入曲线xy坐标的csv文件
file_list = ['./roc_array_code7.csv',
             './roc_array_code3.csv']##########################

data_list_list=[]
for file in file_list:
    tem = []
    with open(file,'r') as csvfile:
        reader = csv.reader(csvfile)
        for row in reader:
            tem.append(row)
        data_list_list.append(tem)

labels=['net7','net3']##########################
counter = 0
for data in data_list_list:
    x=[]
    y=[]
    for row in data:
        x.append(float(row[0]))
        y.append(float(row[1]))
        print(counter)
    plt.plot(x,y,label=labels[counter])
    counter +=1

plt.title('roc')##############################
plt.xlabel('FPR')##############################
plt.ylabel('TPR')##############################
print('2')
plt.legend()
print('3')
plt.savefig('./roc_multi.jpg')##############################

 

问题:代码读取csv文件中的数据,然后画出线图,保存时程序没反应

解决:读取csv文件时,数字是以字符串的形式存在的,画图时,要float(row【0】)转化成数字才能画图,否则保存时会卡住

猜你喜欢

转载自www.cnblogs.com/baitian963/p/12435707.html