今天,学习了利用Python绘制“霍兰德人格分析雷达图”,代码是直接从网课上复制过来的,但是运行的时候出错了。之后上网查了一些资料,最终也是解决了。
网课上的源代码。
#HollandRadarDraw
import numpy as np
import matplotlib.pyplot as plt
import matplotlib
matplotlib.rcParams['font.family']='SimHei'
radar_labels = np.array(['研究型(I)','艺术型(A)','社会型(S)',\
'企业型(E)','常规型(C)','现实型(R)']) #雷达标签
nAttr = 6
data = np.array([[0.40, 0.32, 0.35, 0.30, 0.30, 0.88],
[0.85, 0.35, 0.30, 0.40, 0.40, 0.30],
[0.43, 0.89, 0.30, 0.28, 0.22, 0.30],
[0.30, 0.25, 0.48, 0.85, 0.45, 0.40],
[0.20, 0.38, 0.87, 0.45, 0.32, 0.28],
[0.34, 0.31, 0.38, 0.40, 0.92, 0.28]]) #数据值
data_labels = ('艺术家', '实验员', '工程师', '推销员', '社会工作者','记事员')
angles = np.linspace(0, 2*np.pi, nAttr, endpoint=False)
data = np.concatenate((data, [data[0]]))
angles = np.concatenate((angles, [angles[0]]))
fig = plt.figure(facecolor="white")
plt.subplot(111, polar=True)
plt.plot(angles,data,'o-', linewidth=1, alpha=0.2)
plt.fill(angles,data, alpha=0.25)
plt.thetagrids(angles*180/np.pi, radar_labels,frac = 1.2)
plt.figtext(0.52, 0.95, '霍兰德人格分析', ha='center', size=20)
legend = plt.legend(data_labels, loc=(0.94, 0.80), labelspacing=0.1)
plt.setp(legend.get_texts(), fontsize='large')
plt.grid(True)
plt.savefig('holland_radar.jpg')
plt.show()
运行时候出现了错误。
这个错误是plt.thetagrids(angles*180/np.pi, radar_labels,frac = 1.2),在最新的python中,库的函数有一些参数发生了一定的变化,比如在plt.thetagrids函数中,没有frac参数了,只需将frac删除即可。
此错误的出现原因是角度和数据都闭合了,但是雷达图上的标签没有闭合,这里有两种解决方法。
第一种。
radar_labels = np.array(['研究型(I)','艺术型(A)','社会型(S)','企业型(E)','常规型(C)','现实型(R)','研究型(I)']) #雷达标签
第二种。
data = np.concatenate((data, [data[0]]))
angles = np.concatenate((angles, [angles[0]]))
radar_labels = np.concatenate((radar_labels,[radar_labels[0]]))
如此操作,就可以得到雷达图了。
补充:
貌似直接将这两句注释掉,也可直接运行。
#data = np.concatenate((data, [data[0]]))
#angles = np.concatenate((angles, [angles[0]]))
附录:
Python绘制雷达图时遇到的坑的解决 - 自学编程网 (zxbcw.cn)
照抄霍兰德人格分析雷达图代码也会错 (ˉ▽ˉ;)..._plt.thetagrids()_weixin_50066116的博客-CSDN博客