温馨提示×

Matplotlib中怎么实现热力图

小亿
114
2024-05-13 13:42:14
栏目: 编程语言

要在Matplotlib中绘制热力图,可以使用imshow()函数。以下是一个简单的示例代码:

import numpy as np
import matplotlib.pyplot as plt

data = np.random.rand(5, 5)  # 生成随机数据

plt.imshow(data, cmap='hot', interpolation='nearest')  # 绘制热力图
plt.colorbar()  # 添加颜色条
plt.show()

在这个例子中,我们生成了一个随机的5x5的数据矩阵,然后使用imshow()函数绘制热力图,指定使用hot颜色映射,并设置插值方法为nearest。最后通过colorbar()函数添加颜色条。运行代码后,将会显示出一个热力图。

0