numpy数组中冒号[:,:,0]与[...,0]的区别

先随机产生一个[3,4,5]的numpy数组。则该x维度是3,shape是(3,4,5),总共包含60个元素。

x[:,:,0] 意思是对数组x切片,可以想象成一个正方体数据,每次切下一个面的数据。第二维取0则得出来[3,4]大小的数组,即

 可以验证:

那么[...,0]代表了什么?

首先...只能出现一次,就是说你可以,[ : , : , : ],但是[ ... , ...]就会报错。

使用了 ... 之后,数字0不再是元素的index 了 , 而是 轴(axis)。下面通过numpy.amax()(选出轴最大的元素)来具体说明。

 x大小为(3,4,5)

首先看axis=2,第一个数字189是从x[ ][ ] [0] 到 x[ ][ ] [4] 比较而得,因此一共有3*4=12元素

axis=1,第一个数字99是从x[ ][0] [ ] 到 x[ ][3] [ ] 比较而得,因此一共有3*5=15元素

同理,axis=0,第二个数字189是从x[0] [ ] [ ] 到 x[2] [ ] [ ] 比较而得,因此一共有4*5=20元素

axis=0时 比较的示意图:

最直观的:函数所选的axis的值,就表明 x[ ][ ][ ] 的第几个方块号,从0开始,代表第一个[ ],即x[ ] [ ] [ ],所以维度与axis的对应关系为,对于维度为(3,4,5)的数组,axis=0 长度为3,axis=1长度为4,axis=2长度为5。

则[...,0]表示,与[:,:,0]等价:

同时,还可以这样用。

猜你喜欢

转载自blog.csdn.net/qq_39237205/article/details/124201701
今日推荐