温馨提示×

Matplotlib怎么制作矩阵或二维数组的热力图

小亿
95
2024-05-21 11:22:21
栏目: 编程语言

要制作矩阵或二维数组的热力图,可以使用Matplotlib中的imshow函数。下面是一个简单的示例代码:

import numpy as np
import matplotlib.pyplot as plt

# 创建一个随机的二维数组
data = np.random.rand(10, 10)

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

在这个示例中,我们首先使用numpy库生成一个随机的10x10的二维数组,然后使用imshow函数绘制热力图。参数cmap用于指定颜色映射,interpolation用于指定插值方法,plt.colorbar()用于添加颜色条。最后调用plt.show()显示热力图。您也可以根据自己的需要调整数据以及参数来生成不同的热力图。

0