matplotlib画三维热力图

要画三维热力图，可以使用`mplot3d`子库。具体步骤如下：

1. 导入`mplot3d`和`pyplot`子库：

from mpl_toolkits import mplot3d
import matplotlib.pyplot as plt

2. 创建三维图形对象`fig`和`ax`：

fig = plt.figure()
ax = plt.axes(projection='3d')

3. 绘制三维热力图：

ax.plot_surface(X, Y, Z, cmap='viridis')

其中，`X`、`Y`、`Z`是三个数组，表示三维空间中的点坐标和对应的颜色值。

完整代码示例：

from mpl_toolkits import mplot3d
import matplotlib.pyplot as plt
import numpy as np

# 定义三维空间中的点坐标和对应的颜色值
def f(x, y):
    return np.sin(np.sqrt(x ** 2 + y ** 2))

x = np.linspace(-6, 6, 30)
y = np.linspace(-6, 6, 30)

X, Y = np.meshgrid(x, y)
Z = f(X, Y)

# 创建三维图形对象
fig = plt.figure()
ax = plt.axes(projection='3d')

# 绘制三维热力图
ax.plot_surface(X, Y, Z, cmap='viridis')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')

plt.show()

运行结果如下图所示：