如何绘制多维数组？

import numpy as np
import matplotlib.pyplot as plt

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

# 创建一个绘图画布
fig, ax = plt.subplots()

# 绘制二维数组
for i in range(array_2d.shape[0]):
    for j in range(array_2d.shape[1]):
        ax.text(j, i, array_2d[i, j], ha='center', va='center', color='black')

# 设置坐标轴范围和标签
ax.set_xlim([-0.5, array_2d.shape[1]-0.5])
ax.set_ylim([-0.5, array_2d.shape[0]-0.5])
ax.set_xticks(np.arange(array_2d.shape[1]))
ax.set_yticks(np.arange(array_2d.shape[0]))
ax.set_xticklabels(np.arange(array_2d.shape[1])+1)
ax.set_yticklabels(np.arange(array_2d.shape[0])+1)
ax.xaxis.tick_top()

# 添加网格线
ax.grid(color='black', linestyle='-', linewidth=1)

# 显示绘图结果
plt.show()