# -*- 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】)轉化成數字才能畫圖,否則保存時會卡住