最近需要將實驗數據畫圖出來,由於使用python進行實驗,自然使用到了matplotlib來作圖。
下面的代碼可以作為畫圖的模板代碼,代碼中有詳細注釋,可根據需要進行更改。
1 # -*- coding: utf-8 -*- 2 import numpy as np 3 import matplotlib.pyplot as plt 4 plt.rcParams['font.sans-serif']=['Arial'] #如果要顯示中文字體,則在此處設為:SimHei 5 plt.rcParams['axes.unicode_minus']=False #顯示負號 6 7 x = np.array([3,5,7,9,11,13,15,17,19,21]) 8 A = np.array([0.9708, 0.6429, 1, 0.8333, 0.8841, 0.5867, 0.9352, 0.8000, 0.9359, 0.9405]) 9 B= np.array([0.9708, 0.6558, 1, 0.8095, 0.8913, 0.5950, 0.9352, 0.8000, 0.9359, 0.9419]) 10 C=np.array([0.9657, 0.6688, 0.9855, 0.7881, 0.8667, 0.5952, 0.9361, 0.7848, 0.9244, 0.9221]) 11 D=np.array([0.9664, 0.6701, 0.9884, 0.7929, 0.8790, 0.6072, 0.9352, 0.7920, 0.9170, 0.9254]) 12 13 #label在圖示(legend)中顯示。若為數學公式,則最好在字符串前后添加"$"符號 14 #color:b:blue、g:green、r:red、c:cyan、m:magenta、y:yellow、k:black、w:white、、、 15 #線型:- -- -. : , 16 #marker:. , o v < * + 1 17 plt.figure(figsize=(10,5)) 18 plt.grid(linestyle = "--") #設置背景網格線為虛線 19 ax = plt.gca() 20 ax.spines['top'].set_visible(False) #去掉上邊框 21 ax.spines['right'].set_visible(False) #去掉右邊框 22 23 plt.plot(x,A,color="black",label="A algorithm",linewidth=1.5) 24 plt.plot(x,B,"k--",label="B algorithm",linewidth=1.5) 25 plt.plot(x,C,color="red",label="C algorithm",linewidth=1.5) 26 plt.plot(x,D,"r--",label="D algorithm",linewidth=1.5) 27 28 group_labels=['dataset1','dataset2','dataset3','dataset4','dataset5',' dataset6','dataset7','dataset8','dataset9','dataset10'] #x軸刻度的標識 29 plt.xticks(x,group_labels,fontsize=12,fontweight='bold') #默認字體大小為10 30 plt.yticks(fontsize=12,fontweight='bold') 31 plt.title("example",fontsize=12,fontweight='bold') #默認字體大小為12 32 plt.xlabel("Data sets",fontsize=13,fontweight='bold') 33 plt.ylabel("Accuracy",fontsize=13,fontweight='bold') 34 plt.xlim(3,21) #設置x軸的范圍 35 #plt.ylim(0.5,1) 36 37 #plt.legend() #顯示各曲線的圖例 38 plt.legend(loc=0, numpoints=1) 39 leg = plt.gca().get_legend() 40 ltext = leg.get_texts() 41 plt.setp(ltext, fontsize=12,fontweight='bold') #設置圖例字體的大小和粗細 42 43 plt.savefig('D:\\filename.svg',format='svg') #建議保存為svg格式,再用inkscape轉為矢量圖emf后插入word中 44 plt.show()
下面是上面代碼繪制的圖例:
建議保存圖片的格式svg(因為matplotlib存為eps矢量圖時候會有問題),然后使用inkscape軟件將svg格式轉為emf矢量圖格式。如果svg圖片很多,可以在windows下使用批處理(安裝inkscape軟件后要記得設置path路徑),下面這段代碼能將它所在目錄下的svg文件轉為emf文件。將下面代碼復制到文本文件,改后綴名為bat。
1 @echo off 2 for %%i in (*.svg) do ( 3 echo %%i 4 inkscape -f %%i -M %%~ni.emf 5 ) 6 @echo Finished