软件测试 您所在的位置:网站首页 python使用matplotlib绘图 软件测试

软件测试

#软件测试| 来源: 网络整理| 查看: 265

前言

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

当然,只要我们提供数据,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')

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

在这里插入图片描述

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

总结

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



【本文地址】

公司简介

联系我们

今日新闻

    推荐新闻

    专题文章
      CopyRight 2018-2019 实验室设备网 版权所有