import matplotlib.pyplot as pltimport numpy as np# 生成随机整数数组arr = np.random.randint(0,90, size=(20, 20))# 绘制热力图并显示数值fig, ax = plt.subplots()im = ax.imshow(arr)#, cmap='hot')# 显示数值for i in range(len(arr)):for j in range(len(arr[0])):text = ax.text(j, i, arr[i, j],ha="center", va="center", color="w")# 添加颜色条plt.colorbar(im)# 显示图形plt.show()
import matplotlib.pyplot as pltimport numpy as np# 构造数据x = np.linspace(-2, 2, 100)y = np.linspace(-2, 2, 100)X, Y = np.meshgrid(x, y)Z = np.exp(-X**2 - Y**2)# 绘制连续热力图plt.contourf(X, Y, Z, cmap='hot')# 添加颜色条plt.colorbar()# 显示图形plt.show()