软件测试|如何用Python绘制雷达图

2023-03-05 10:38:45 浏览数 (1)

前言

我有朋友问我,他准备买车,预算20-25万,他在考虑几个车,说现在很难做出决定,让我帮他参谋参谋,该买哪个?

我只能问,你考虑的是哪几款车?

我朋友提出了下列的名单,帕萨特,迈腾,凯美瑞,亚洲龙,天籁,雅阁。

说实话,这6款车都不错,是中级车市场的主流车型,没想到这小子混的是真不错。

那么我们应该如何提出我们的意见,我们在提出我们的意见之前,需要对各个车型做出对比,选出最符合我朋友要求的车型。那么指标这么多,我们要如何更直观的展示出来各个车型的优缺点,更便于他去选择,我决定绘制一个雷达图来给他展示这些各个车型的优缺点。

数据准备

选择买哪款车,我们主要考虑的有如下指标:舒适性,油耗,性能,安全,操控,这就开始干,根据我们在汽车之家上查到的数据给各个车型赋予分数,如下:

代码语言:python代码运行次数:0复制
# 分别为帕萨特,迈腾,凯美瑞,亚洲龙,天籁,雅阁
stats = [
    {"舒适性": 85, "油耗": 75, "性能": 90, "安全": 70, "操控": 90},
    {"舒适性": 87, "油耗": 72, "性能": 95, "安全": 75, "操控": 95},
    {"舒适性": 80, "油耗": 95, "性能": 70, "安全": 85, "操控": 80},
    {"舒适性": 85, "油耗": 90, "性能": 75, "安全": 85, "操控": 75},
    {"舒适性": 85, "油耗": 85, "性能": 80, "安全": 80, "操控": 85},
    {"舒适性": 75, "油耗": 90, "性能": 82, "安全": 85, "操控": 88},

]

绘制单车型雷达图

我们先绘制出单个车型的雷达图,比如我们要先绘制帕萨特的雷达图,我们的代码如下:

代码语言:python代码运行次数:0复制
import numpy as np
import matplotlib.pyplot as plt

plt.rcParams['font.sans-serif'] = ['SimHei']  # 显示中文
plt.rcParams['axes.unicode_minus'] = False  # 正常显示负号

stats = [
    {"舒适性": 85, "油耗": 75, "性能": 90, "安全": 70, "操控": 90},
    {"舒适性": 87, "油耗": 72, "性能": 95, "安全": 75, "操控": 95},
    {"舒适性": 80, "油耗": 95, "性能": 70, "安全": 85, "操控": 80},
    {"舒适性": 85, "油耗": 90, "性能": 75, "安全": 85, "操控": 75},
    {"舒适性": 85, "油耗": 85, "性能": 80, "安全": 80, "操控": 85},
    {"舒适性": 75, "油耗": 90, "性能": 82, "安全": 85, "操控": 88},

]
data_length = len(stats[0])
# 将极坐标根据数据长度进行等分
angles = np.linspace(0, 2 * np.pi, data_length, endpoint=False)
labels = [key for key in stats[0].keys()]
score = [[v for v in stat.values()] for stat in stats]
# 使雷达图数据封闭
score_a = np.concatenate((score[0], [score[0][0]]))
angles = np.concatenate((angles, [angles[0]]))
labels = np.concatenate((labels, [labels[0]]))
# 设置图形的大小
fig = plt.figure(figsize=(9, 9), dpi=100)
# 新建一个子图
ax = plt.subplot(111, polar=True)
# 绘制雷达图
ax.plot(angles, score_a, color='g')
# 设置雷达图中每一项的标签显示
ax.set_thetagrids(angles * 180 / np.pi, labels)
# 设置雷达图的0度起始位置
ax.set_theta_zero_location('N')  # E W S N SW SE NW NE
# 设置雷达图的坐标刻度范围
ax.set_rlim(0, 100)
# 设置雷达图的坐标值显示角度,相对于                                                                                                                                                                                                                                                                                                                                                                                            y               起始角度的偏移量
ax.set_rlabel_position(270)
ax.set_title("车型对比")
plt.legend(["帕萨特"], loc='best')
plt.show()

绘制出的雷达图如下所示:

在这里插入图片描述在这里插入图片描述

绘制全车型的雷达图

上方绘制单车型的雷达图我们可以知道,我们只读取了一个车型的数据,并且只绘制了一个车型的数据,我们现在要绘制上述6个车型的雷达图,需要读取全部6个车型的数据,并且绘制6个车型的雷达图,代码如下:

代码语言:python代码运行次数:0复制
import numpy as np
import matplotlib.pyplot as plt

plt.rcParams['font.sans-serif'] = ['SimHei']  # 显示中文
plt.rcParams['axes.unicode_minus'] = False  # 正常显示负号

stats = [
    {"舒适性": 85, "油耗": 75, "性能": 90, "安全": 70, "操控": 90},
    {"舒适性": 87, "油耗": 72, "性能": 95, "安全": 75, "操控": 95},
    {"舒适性": 80, "油耗": 95, "性能": 70, "安全": 85, "操控": 80},
    {"舒适性": 85, "油耗": 90, "性能": 75, "安全": 85, "操控": 75},
    {"舒适性": 85, "油耗": 85, "性能": 80, "安全": 80, "操控": 85},
    {"舒适性": 75, "油耗": 90, "性能": 82, "安全": 85, "操控": 88},

]
data_length = len(stats[0])
# 将极坐标根据数据长度进行等分
angles = np.linspace(0, 2 * np.pi, data_length, endpoint=False)
labels = [key for key in stats[0].keys()]
score = [[v for v in stat.values()] for stat in stats]
# 使雷达图数据封闭
score_a = np.concatenate((score[0], [score[0][0]]))
score_b = np.concatenate((score[1], [score[1][0]]))
score_c = np.concatenate((score[2], [score[2][0]]))
score_d = np.concatenate((score[3], [score[3][0]]))
score_e = np.concatenate((score[4], [score[4][0]]))
score_f = np.concatenate((score[5], [score[5][0]]))
angles = np.concatenate((angles, [angles[0]]))
labels = np.concatenate((labels, [labels[0]]))
# 设置图形的大小
fig = plt.figure(figsize=(9, 9), dpi=100)
# 新建一个子图
ax = plt.subplot(111, polar=True)
# 绘制雷达图
ax.plot(angles, score_a, color='g')
ax.plot(angles, score_b, color='b')
ax.plot(angles, score_c, color='c')
ax.plot(angles, score_d, color='m')
ax.plot(angles, score_e, color='y')
ax.plot(angles, score_f, color='r')

# 设置雷达图中每一项的标签显示
ax.set_thetagrids(angles * 180 / np.pi, labels)
# 设置雷达图的0度起始位置
ax.set_theta_zero_location('N')  # E W S N SW SE NW NE
# 设置雷达图的坐标刻度范围
ax.set_rlim(0, 100)
# 设置雷达图的坐标值显示角度,相对于                                                                                                                                                                                                                                                                                                                                                                                            y               起始角度的偏移量
ax.set_rlabel_position(270)
ax.set_title("车型对比")
plt.legend(["帕萨特", "迈腾", "凯美瑞", "亚洲龙", "天籁", "雅阁"], loc='best')
plt.show()

绘制的图像如下:

在这里插入图片描述在这里插入图片描述

进一步优化

我们可以看到,上面绘制的雷达图,存在线条过于紧密的问题,让人眼花缭乱,我们能不能使得绘制的图不再那么紧密,显而易见,没有哪款车型的某项得分在65分以下,我们是否可以将刻度从65分开始呢,答案是可以的,我们只需要修改一下这一行代码即可,代码如下:

代码语言:python代码运行次数:0复制
# 设置雷达图的坐标刻度范围
ax.set_rlim(65, 100)

修改代码后重新绘制的雷达图如下:

在这里插入图片描述在这里插入图片描述

我们可以看到,不同车型的差距变的更直观许多。

总结

我们本篇文章主要借着替我朋友分析买哪款车好的契机,向大家介绍了使用matplotlib绘制雷达图的方法,包括对于雷达图的优化,下一篇文章,我们将介绍使用pyecharts库来绘制雷达图,看看是否会更加便捷。最后,我朋友家拆迁了,并没有买这么便宜的车型,他直接买了我的梦想——保时捷帕纳梅拉。

0 人点赞