python 多个雷达图

原创

mob64ca12d52440 2023-10-20 08:29:55 ©著作权

文章标签 数据 Python python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12d52440的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python 多个雷达图

雷达图是一种常用的数据可视化方法，它可以同时展示多个变量的数值，帮助我们更直观地了解数据之间的关系和差异。Python提供了多种库和工具，方便我们绘制雷达图，本文将介绍如何使用Python绘制多个雷达图，并展示具体的代码示例。

准备工作

在开始编写代码之前，首先需要安装一些必要的库。我们将使用matplotlib库来绘制雷达图，使用numpy库来进行数据处理。可以使用以下命令进行安装：

pip install matplotlib
pip install numpy

绘制单个雷达图

在绘制多个雷达图之前，我们先来了解如何绘制单个雷达图。下面是一个简单的示例代码：

import numpy as np
import matplotlib.pyplot as plt

# 数据
categories = ['A', 'B', 'C', 'D', 'E']
values = [4, 3, 5, 2, 4]

# 参数设置
N = len(categories)
angles = np.linspace(0, 2 * np.pi, N, endpoint=False).tolist()
values += values[:1]
angles += angles[:1]

# 绘制雷达图
fig, ax = plt.subplots(figsize=(6, 6), subplot_kw=dict(polar=True))
ax.fill(angles, values, color='blue', alpha=0.25)
ax.set_xticks(angles[:-1])
ax.set_xticklabels(categories)
ax.set_yticks([1, 2, 3, 4, 5])
ax.set_yticklabels([1, 2, 3, 4, 5])
ax.set_title('Radar Chart')

plt.show()

运行以上代码，将得到一个包含5个变量的雷达图。代码中，我们首先定义了数据的分类和数值。然后设置雷达图的参数，包括每个变量的角度和数值。最后使用fill函数绘制雷达图，并设置相关的标签和标题。

绘制多个雷达图

要绘制多个雷达图，我们可以使用循环语句来遍历多组数据，并在每次循环中绘制一个雷达图。下面是一个示例代码，展示如何绘制3个雷达图：

import numpy as np
import matplotlib.pyplot as plt

# 数据
categories = ['A', 'B', 'C', 'D', 'E']
data = [[4, 3, 5, 2, 4], [3, 2, 4, 3, 1], [5, 1, 3, 4, 2]]

# 参数设置
N = len(categories)
angles = np.linspace(0, 2 * np.pi, N, endpoint=False).tolist()
angles += angles[:1]

# 绘制多个雷达图
fig, axs = plt.subplots(1, 3, figsize=(18, 6), subplot_kw=dict(polar=True))

for i, ax in enumerate(axs):
    values = data[i]
    values += values[:1]
    ax.fill(angles, values, color='blue', alpha=0.25)
    ax.set_xticks(angles[:-1])
    ax.set_xticklabels(categories)
    ax.set_yticks([1, 2, 3, 4, 5])
    ax.set_yticklabels([1, 2, 3, 4, 5])
    ax.set_title(f'Radar Chart {i+1}')

plt.show()

运行以上代码，将得到一个包含3个雷达图的图表。代码中，我们定义了一个包含3组数据的列表，并通过循环语句遍历每组数据，绘制一个雷达图。每个雷达图的相关参数和标签设置与单个雷达图相同。