numpy中ndim

建立一个np.array

data2 = [[1, 2, 3, 4], [5, 6, 7, 8]]
arr2 = np.array(data2)
arr2

array([[1, 2, 3, 4],
       [5, 6, 7, 8]])

ndim查看数组的维数

arr2.ndim#维度是2

2

np.ones((2, 3, 2))#先是一组2*1 然后3个2*1再成一组 然后2个3个2*1再成一组

array([[[1., 1.],
        [1., 1.],
        [1., 1.]],

       [[1., 1.],
        [1., 1.],
        [1., 1.]]])

np.ones((2, 3, 2)).ndim

3

np.ones((2, 3, 2, 3))

np.ones((2, 3, 2, 3)).ndim

4

如何判断数组维度

有几个[ ]就是几维
np.ones(1)
array([1.])[1] 一维

np.ones((1,2))
array([[1., 1.]])二维