如何在phython中以3d形式表示2d数组？

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

# 创建一个2D数组
array_2d = np.array([[1, 2, 3],
                     [4, 5, 6],
                     [7, 8, 9]])

# 获取数组的行数和列数
rows, cols = array_2d.shape

# 创建一个网格
x = np.arange(cols)
y = np.arange(rows)
X, Y = np.meshgrid(x, y)

# 将2D数组转换为3D形式
Z = array_2d

# 创建3D图形对象
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 绘制3D图形
ax.plot_surface(X, Y, Z)

# 设置坐标轴标签
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')

# 显示图形
plt.show()