matplotlib绘制符合论文要求的图片

　　最近需要将实验数据画图出来，由于使用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()

更多相关文章

随机推荐