A-A+
Python NumPy数组维数显示函数ndim和shape
在Python NumPy中数组的维数可以通过 np.dim()函数获得。数组的形状可以通过实例变量 shape获得,shape的结果是个元组(tuple)。
示例代码:
>>> A=np.array([[1,2,3,4,5],[2,3,4,5,6],[3,4,5,6,7],[4,5,6,7,8]])>>> np.ndim(A)2>>> B=np.zeros((2,3,4))>>> B.ndim3>>> print(B)[[[0. 0. 0. 0.] [0. 0. 0. 0.] [0. 0. 0. 0.]] [[0. 0. 0. 0.] [0. 0. 0. 0.] [0. 0. 0. 0.]]]>>> A.shape(4, 5)>>> B.shape(2, 3, 4)>>> C=np.array([1,2,3,4,5])>>> C.ndim1>>> C.shape(5,)