python 画热度图

热度图是一种在数据可视化中常常使用到的图表类型，它可以用来显示数据在二维平面上的关联程度。Python 的 Matplotlib 库提供了绘制热度图的功能，并且使用简单。 要使用 Matplotlib 绘制热度图，我们需要使用其子模块 pyplot，并且使用 imshow() 函数。下面是一个简单的代码示例：

import numpy as np
import matplotlib.pyplot as plt
# 生成随机矩阵
data = np.random.rand(5, 5)
plt.imshow(data, cmap=plt.cm.hot, interpolation='nearest')
plt.colorbar()
plt.show()

在这个示例中，我们首先使用 NumPy 库生成了一个 $5\times 5$ 的随机矩阵，然后使用 imshow() 函数将这个矩阵绘制成热度图。其中 cmap 参数指定了使用的色彩映照，interpolation 参数指定了数据的差值方式。 除此以外，我们还使用了 colorbar() 函数添加了一个色彩条，方便查看图表对应的数值大小。 最后再使用 show() 函数将绘制好的图表显示出来。运行该程序，我们可以得到一个类似以下的热度图：

![热度图示例](/upload/allimg/2023/06/22/5q51q4twmit.png)

从这个示例中我们可以看出，Python 绘制热度图非常简单，只需要使用 Matplotlib 库的 imshow() 函数便可。固然，在实际利用中，我们可能还需要对图表进行一些美化，比如添加标题、轴标签、刻度等等，这些都可以通过 Matplotlib 提供的函数轻松实现。