radar chart显示的是不同数值类型变量对于一个或多个entities的影响。

在Matplotlib中并没有内建函数支持雷达图,因此需要自己实现。

首先,引入类库

import matplotlib.pyplot as plt
import pandas as pd
from math import pi

创建数据集

df = pd.DataFrame({
'group': ['A','B','C','D'],
'var1': [38, 1.5, 30, 4],
'var2': [29, 10, 9, 34],
'var3': [8, 39, 23, 24],
'var4': [7, 31, 33, 14],
'var5': [28, 15, 32, 14]
})

获取特征变量

categories=list(df)[1:]
N = len(categories)

对数据集的第一列数据绘制雷达图,因此

values=df.loc[0].drop('group').values.flatten().tolist()
values += values[:1]
values

求取特征值的角度

angles = [n / float(N) * 2 * pi for n in range(N)]
angles += angles[:1]

初始化spider plot

ax = plt.subplot(111, polar=True)

绘制x标签

plt.xticks(angles[:-1], categories, color='grey', size=8)

绘制y标签

ax.set_rlabel_position(0)
plt.yticks([10,20,30], ["10","20","30"], color="grey", size=7)
plt.ylim(0,40)

引入数据

ax.plot(angles, values, linewidth=1, linestyle='solid')

填充区域

ax.fill(angles, values, 'b', alpha=0.1)

results matching ""

    No results matching ""