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


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM