ndim 与 shape的区别

[[ 1., 0., 0.],
 [ 0., 1., 2.]] 

在上面这个例子中,数组的ndim为2(它有两个维度(简单的辨别两层方括号)).

第一个维度的长度为2,也就是有两个子数组

第二个维度的长度为3,也就是每个子数组里有三个元素

所以shape属性返回的是(2,3)

简单来看 shape就是每个 维度中的元素/单元 长度size

ndim就是每个数组的嵌套层数

下面是一个二维数组,ndim为2,shape属性值为(3,3)

b = 
[
   [1,2,3],
   [4,5,6],
   [7,8,9]
]
取得一个确切的元素,用b[i][j]的格式,数组嵌套了2层,第一层有三个元素,第二层也是三个元素

下面是一个三维数组,ndim为3,shape属性值为(2,2,3)

c = 
[
    [
        [1,2,3],
        [4,5,6]
    ],
    [
       [7,8,9],
       [10,11,12]
    ]
]
取得一个确切的元素,
用b[i][j][k]的格式,数组嵌套了3层,第一层有2个元素,第二层也是2个元素,第三层有三个元素

或者 ndim = len(shape)

axis = 0表示第一层layer

在代码中看到的效果就是b从外向里数第一层 [ ],对应的元素(length = 3)就是第一层 [ ] 中用逗号分隔的全部元素(注意:不要关注第二层及以上 [ ] 里的逗号),很明显这一层的元素数目为3;

axis = 1表示第二层layer:第一层layer中用逗号分隔开的任意一个 [ ] 中,再由逗号分隔开的全部元素(length = 2);

axis = 2同理,表示最后一层layer中的元素(length = 4)。这么说还是有些乱,我把刚才的array重新布置一下:

黑色代表第一层layer,可以看到有三个元素,红色代表第二层layer,在第一层layer相同的前提下有两个元素,蓝色代表第三层layer,在第一层和第二层layer都相同的前提下有4个元素。这样也就对应上了b.shape = (3, 2, 4)的结果。

这时候如果我们想定位到元素8应该怎么索引?可以看到8在第一层layer的第二个元素内,同时在第二层layer的第二个元素内、第三层layer的第一个元素,也就是说:8是axis = 0轴上的第二个位置、axis = 1轴上的第二个位置,axis = 2的第一个位置共同retrieve到的元素。

8的索引

[1,1,0]

详细参考

https://zhuanlan.zhihu.com/p/25761406

原文地址:https://www.cnblogs.com/clemente/p/10399319.html