Input z must be 2D, not 3D

这个错误通常是因为 `contour` 函数的输入数据 `Z` 是一个三维数组，而 `contour` 函数只能处理二维数组。因此，需要将 `Z` 降到二维数组。

一种解决方法是使用 numpy 库中的 `np.max` 或 `np.min` 函数将 `Z` 沿着某一维度降维，例如：

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

# 生成数据
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)

# 将 Z 降到二维数组
Z_2d = np.max(Z, axis=0)  # 或者使用 np.min 函数

# 绘制图形
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
ax.contour(X, Y, Z_2d)
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')
plt.show()

在这个例子中，我们使用 `np.max` 函数将 `Z` 沿着第一个维度降维，得到一个二维数组 `Z_2d`，然后将其传递给 `contour` 函数进行绘图。

需要注意的是，降维过程可能会破坏数据的结构。因此，在进行数据降维时，需要综合考虑数据的特点和分析需求，选择合适的方法。