要获取Numpy数组的维度,可以使用ndarray.ndim
属性,其中ndarray
是Numpy数组的变量名。例如:
import numpy as np
# 创建一个2x3的二维数组
a = np.array([[1, 2, 3], [4, 5, 6]])
# 获取数组的维度
dim = a.ndim
print(dim) # 输出: 2
这样,变量dim
就储存了数组a
的维度。
你也可以使用shape
属性获取数组的形状,它返回一个元组,表示数组在每个维度上的大小。例如: