[python] 基于matplotlib实现雷达图的绘制

绘制雷达图（又称为蜘蛛网图或星形图）在数据可视化中非常有用，特别是当你需要展示多个变量的比较时。在Python中，Matplotlib库可以用来绘制雷达图。下面是一个基于Matplotlib的示例代码，展示了如何绘制雷达图。

首先，确保你已经安装了Matplotlib库。如果没有安装，可以使用以下命令进行安装：


pip install matplotlib

以下是一个完整的示例代码，展示了如何绘制雷达图：

import numpy as np
import matplotlib.pyplot as plt
from math import pi# 示例数据：五个变量的值
labels = ['A', 'B', 'C', 'D', 'E']
values = [4, 3, 2, 5, 4]# 数据点的数量
num_vars = len(labels)# 计算角度
angles = [n / float(num_vars) * 2 * pi for n in range(num_vars)]
angles += angles[:1]  # 闭合图形# 初始化数据
values += values[:1]  # 闭合图形# 创建一个新的图形
fig, ax = plt.subplots(figsize=(6, 6), subplot_kw=dict(polar=True))# 绘制雷达图
ax.fill(angles, values, color='b', alpha=0.25)
ax.plot(angles, values, color='b', linewidth=2)# 填充标签
ax.set_xticks(angles[:-1])
ax.set_xticklabels(labels)# 设置y轴的限制
ax.set_ylim(0, max(values) + 1)# 添加网格线
ax.yaxis.grid(True)
ax.xaxis.grid(True)# 显示图形
plt.show()

代码解释

导入必要的库：
- numpy 用于数值计算（这里其实没有用到，但通常很有用）。
- matplotlib.pyplot 用于绘图。
- math.pi 用于计算角度（虽然可以直接用 np.pi）。
定义数据和标签：
- labels 定义了每个变量的标签。
- values 定义了每个变量的值。
计算角度：
- 使用 num_vars 计算每个变量在雷达图上的角度。
- 闭合图形：通过 angles += angles[:1] 将最后一个角度与第一个角度连接起来，形成一个闭合的图形。
初始化数据：
- 通过 values += values[:1] 将最后一个值与第一个值连接起来，以闭合图形。
创建图形：
- 使用 plt.subplots 创建一个极坐标子图。
绘制雷达图：
- 使用 ax.fill 填充雷达图的区域。
- 使用 ax.plot 绘制雷达图的线条。
设置标签和限制：
- 使用 ax.set_xticks 和 ax.set_xticklabels 设置x轴的刻度标签。
- 使用 ax.set_ylim 设置y轴的限制。
添加网格线：
- 使用 ax.yaxis.grid 和 ax.xaxis.grid 添加网格线。
显示图形：
- 使用 plt.show 显示图形。