要在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()
函数添加颜色条。运行代码后,将会显示出一个热力图。