此内容将带领大家了解matplotlib.pyplot的polar()绘制雷达图函数,且只调用一次plt.polar和plt.fill。
1.1首先我们导入绘图需要的模块
在我们开始前,让我们先导入matplotlib.pyplot和numpy模块,分别将这两个模块重命名为plt和np,这才能使用matplotlib库绘制一个雷达图的图表。
import matplotlib.pyplot as plt
import numpy as np
1.2显示中文
plt.rcParams['font.family'] = 'SimHei'
plt.rcParams['axes.unicode_minus'] = False
需要说明的是,matolotlib默认不支持显示中文,由于条形图的刻度标签是中文文本,因此需要将系统的字体修改为SimHei。
1.3 pyplot的polar()函数的语法格式如下所示:
polar(theta,r,**kwargs)
2.1 准备数据
这是准备的专业三位同学各科目成绩
dim_num = 6
radians = np.linspace(0, 2 * np.pi, dim_num, endpoint=False)
radians = np.concatenate((radians, [radians[0]]))
# A同学: '数学','英语','线代','爬虫','数据可视化','吃喝玩乐' ---> 41,38,36,62,68,100
score_a = np.array([41,38,36,62,68,100])
score_a = np.concatenate((score_a, [score_a[0]]))
# B同学:'数学','英语','线代','爬虫','数据可视化','吃喝玩乐' ---> 91,88,83,72,78,26
score_b = np.array([91,88,83,72,78,26])
score_b = np.concatenate((score_b, [score_b[0]]))
# C同学:'数学','英语','线代','爬虫','数据可视化','吃喝玩乐' ---> 81,78,85,76,74,36
score_c = np.array([81,78,85,76,74,36])
score_c = np.concatenate((score_c, [score_c[0]]))
2.2绘制多边形(雷达图) (只调用一次plt.polar)
plt.polar(radians,score_a,radians,score_b,radians,score_c)
#只调用一次plt.polar函数的话,每一个同学的数据都要进行闭环
2.3设置维度标签
radar_labels = ['数学','英语','线代','爬虫','数据可视化','吃喝玩乐']
radar_labels = np.concatenate((radar_labels, [radar_labels[0]]))
2.4设置极坐标的标签
angles = radians * 180/np.pi # 弧度转角度
plt.thetagrids(angles, labels=radar_labels) # 设置新的刻度标签
2.5填充多边形(只调用一次plt.fill)
plt.fill(radians,score_a,'r',radians,score_b,'b',radians,score_c,'g',alpha=0.3)
填充颜色也是要一个一个传输,不传输系统也会自动填充颜色。
2.6展示图形
plt.show()