python 判断数组维度

在Python中，可以使用numpy库来判断数组的维度。numpy是一个常用的学计算库，提供了对多维数组的支持。

要判断一个数组的维度，可以使用numpy的ndim属性。ndim属性返回数组的维度数。

下面是一个示例代码：

import numpy as np

# 创建一个二维数组
arr_2d = np.array([[1, 2, 3], [4, 5, 6]])
print("arr_2d的维度：", arr_2d.ndim)

# 创建一个三维数组
arr_3d = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print("arr_3d的维度：", arr_3d.ndim)

输出结果为：

arr_2d的维度： 2
arr_3d的维度： 3