Python雷达图|极客笔记

Python雷达图

雷达图（Radar Chart）又称为蜘蛛图（Spider Chart）或者极坐标图（Polar Chart），是一种用于显示多变量数据的图表类型。它的主要特点是可以在同一个图表中展示多个变量的数据，并且能够直观地展示各个变量之间的相对关系。

雷达图通常由多个同心圆和一系列从中心向外部延伸的射线组成。每个变量通常代表一个射线，不同的数据点通过连接各个射线上的点，最终形成一个多边形。这个多边形的形状和大小可以直观地反映出各个变量的数值大小和差异，从而帮助我们更好地理解数据之间的关系。

在本文中，我们将使用Python中的matplotlib库来实现雷达图的绘制。下面我们将逐步介绍如何使用Python绘制雷达图，并通过示例代码演示具体的实现方法。

准备工作

在绘制雷达图之前，我们需要先安装matplotlib库。如果你还没有安装该库，可以通过以下命令使用pip进行安装：

pip install matplotlib

安装完成后，我们就可以开始准备数据并绘制雷达图了。

绘制雷达图

首先，我们需要准备一组数据用于绘制雷达图。在这里，我们假设有五个变量（A、B、C、D、E），它们的取值范围都在0到10之间。我们将使用这些变量的数据来绘制雷达图。

import numpy as np
import matplotlib.pyplot as plt

# 变量个数
labels=np.array(['A', 'B', 'C', 'D', 'E'])
data = np.array([5, 7, 9, 6, 8])

angles=np.linspace(0, 2*np.pi, len(labels), endpoint=False).tolist()

data=np.concatenate((data,[data[0]]))
angles=np.concatenate((angles,[angles[0]])

fig, ax = plt.subplots(figsize=(6, 6), subplot_kw=dict(polar=True))
ax.fill(angles, data, color='skyblue', alpha=0.25)

ax.set_yticks([2, 4, 6, 8, 10])
ax.set_yticklabels(['2', '4', '6', '8', '10'])
plt.show()

以上代码中，我们首先引入numpy和matplotlib.pyplot库，然后定义了五个变量labels和对应的数据data。接着，我们计算了每个变量所处的角度angles，并在最后一个角度之后加入第一个角度，以闭合雷达图。

然后，我们创建了一个雷达图，并使用ax.fill方法填充数据。最后，我们设置了雷达坐标轴的标签和刻度，通过plt.show()方法展示了绘制的雷达图。