热力图绘制热力图 matplotlib

import matplotlib.pyplot as plt
import 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 plt
import 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()