软件测试|教你用matplotlib绘制热力图

时间:2022-01-18 01:17:22

前言

热力图,是一种通过对色块着色来显示数据的统计图表。绘图时,需指定颜色映射的规则。例如,较大的值由较深的颜色表示,较小的值由较浅的颜色表示;较大的值由偏暖的颜色表示,较小的值由较冷的颜色表示,等等。

当然,只要我们提供数据,Python就能够绘制出热力图。下面我们就来介绍一下如何使用Python结合matplotlib来绘制热力图。

环境数据准备

我们使用到的还是matplotlib,所以环境安装还是一条命令如下:

pip install matplotlib

数据我们采取随机生成,随机生成不同车型在不同省份的销售数据,通过numpy读取数据。

代码实现

import numpy as np
import matplotlib as mpl
import matplotlib.pyplot as plt

# 防止乱码
mpl.rcParams["font.sans-serif"] = ["SimHei"]
mpl.rcParams["axes.unicode_minus"] = False

types = ["朗逸", "卡罗拉", "速腾", "雷凌", "思域", "轩逸", "帝豪"]
provices = ['广东', '江西', '浙江', '江苏', '湖北', '上海', '陕西']
sales = np.random.rand(7, 7)*5 # 车型销售数据
plt.rcParams['font.size'] = 18
plt.rcParams['font.weight'] = 'heavy'
plt.figure(figsize=(9,9))
im = plt.imshow(sales)
plt.xticks(np.arange(len(provices)),provices,rotation = 45,ha = 'right')
plt.yticks(np.arange(len(types)),types)

# 绘制⽂本
for i in range(len(types)):
    for j in range(len(provices)):
        text = plt.text(j, i, round(sales[i, j],1),
                        ha="center", va="center", color='r')

plt.title("不同省份各车型销售数据 (in 10000/year)",pad = 20)

plt.savefig('./热⼒图.png')

运行代码绘制的图像如下:

软件测试|教你用matplotlib绘制热力图

从图像上我们可以看到,销量高的单元格颜色越偏暖色调,销量低的单元格颜色越偏冷色调,这就是热力图的。热度越高颜色越深,热度越低颜色越浅。

总结

本文主要讲解了使用Python结合matplotlib绘制热力图的方法,后续我们将介绍使用Python在地图上绘制热力图。