欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

机器学习之matplotlib实例笔记

程序员文章站 2022-10-09 12:13:16
下面的图型是在一幅画布上建立的四个球员相关数据的极坐标图 关于这个图的代码如下: 上述代码也可以利用for循环简化一下 ......

下面的图型是在一幅画布上建立的四个球员相关数据的极坐标图

机器学习之matplotlib实例笔记

 

 关于这个图的代码如下:

 1 #_*_coding:utf-8_*_
 2 import numpy as np
 3 import matplotlib.pyplot as plt
 4 from matplotlib.font_manager import fontproperties
 5 plt.style.use('ggplot')
 6 
 7 font=fontproperties(fname=r'c:\windows\fonts\simsun.ttc',size=12)
 8 # 本行是为了有效显示中文的字体与大小
 9 
10 ability_size=6
11 ability_label=['进攻','防守','盘带','速度','体力','射术']
12 
13 # 创建每个极坐标的位置
14 ax1=plt.subplot(221,projection='polar')
15 ax2=plt.subplot(222,projection='polar')
16 ax3=plt.subplot(223,projection='polar')
17 ax4=plt.subplot(224,projection='polar')
18 
19 player={
20     'm': np.random.randint(size=ability_size, low=60, high=99),
21     'h': np.random.randint(size=ability_size, low=60, high=99),
22     'p': np.random.randint(size=ability_size, low=60, high=99),
23     'q': np.random.randint(size=ability_size, low=60, high=99),
24 }
25 
26 theta=np.linspace(0,2*np.pi,6,endpoint=false)
27 theta=np.append(theta,theta[0])
28 
29 #下面分别画四个球员的能力极坐标图
30 player['m']=np.append(player['m'],player['m'][0])
31 ax1.plot(theta,player['m'],'r')
32 ax1.fill(theta,player['m'],'r',alpha=0.3)
33 ax1.set_xticks(theta)
34 ax1.set_xticklabels(ability_label,fontproperties=font)
35 ax1.set_title('梅西',fontproperties=font,color='r',size=20)
36 ax1.set_yticks([20,40,60,80,100])
37 
38 player['h']=np.append(player['h'],player['h'][0])
39 ax2.plot(theta,player['h'],'g')
40 ax2.fill(theta,player['h'],'g',alpha=0.3)
41 ax2.set_xticks(theta)
42 ax2.set_xticklabels(ability_label,fontproperties=font)
43 ax2.set_title('哈维',fontproperties=font,color='g',size=20)
44 ax2.set_yticks([20,40,60,80,100])
45 
46 player['p']=np.append(player['p'],player['p'][0])
47 ax3.plot(theta,player['p'],'b')
48 ax3.fill(theta,player['p'],'b',alpha=0.3)
49 ax3.set_xticks(theta)
50 ax3.set_xticklabels(ability_label,fontproperties=font)
51 ax3.set_title('皮克',fontproperties=font,color='b',size=20)
52 ax3.set_yticks([20,40,60,80,100])
53 
54 player['q']=np.append(player['q'],player['q'][0])
55 ax4.plot(theta,player['q'],'y')
56 ax4.fill(theta,player['q'],'y',alpha=0.3)
57 ax4.set_xticks(theta)
58 ax4.set_xticklabels(ability_label,fontproperties=font)
59 ax4.set_title('切赫',fontproperties=font,color='y',size=20)
60 ax4.set_yticks([20,40,60,80,100])
61 
62 plt.show()

上述代码也可以利用for循环简化一下

 1 #_*_coding:utf-8_*_
 2 import numpy as np
 3 import matplotlib.pyplot as plt
 4 from matplotlib.font_manager import fontproperties
 5 plt.style.use('ggplot')
 6 
 7 font=fontproperties(fname=r'c:\windows\fonts\simsun.ttc',size=12)
 8 
 9 ability_size=6
10 ability_label=['进攻','防守','盘带','速度','体力','射术']
11 
12 ax1=plt.subplot(221,projection='polar')
13 ax2=plt.subplot(222,projection='polar')
14 ax3=plt.subplot(223,projection='polar')
15 ax4=plt.subplot(224,projection='polar')
16 
17 player={
18     'm': np.random.randint(size=ability_size, low=60, high=99),
19     'h': np.random.randint(size=ability_size, low=60, high=99),
20     'p': np.random.randint(size=ability_size, low=60, high=99),
21     'q': np.random.randint(size=ability_size, low=60, high=99),
22 }
23 
24 theta=np.linspace(0,2*np.pi,6,endpoint=false)
25 theta=np.append(theta,theta[0])
26 
27 color4=['r','b','g','y']
28 player4=['m','h','p','q']
29 ax=[ax1,ax2,ax3,ax4]
30 name=['梅西','哈维','皮克','切赫']
31 for i in range(4):
32     player[player4[i]]=np.append(player[player4[i]],player[player4[i]][0])
33     ax[i].plot(theta,player[player4[i]],color4[i])
34     ax[i].fill(theta,player[player4[i]],color4[i],alpha=0.3)
35     ax[i].set_xticks(theta)
36     ax[i].set_xticklabels(ability_label,fontproperties=font)
37     ax[i].set_title(name[i],fontproperties=font,color=color4[i],size=20)
38     ax[i].set_yticks([20,40,60,80,100])
39 
40 plt.show()