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

今天在看光流法将稠密光流图结果转为bgr可视化图的代码时，遇到了上述情况，写此博客便于理解清晰。

如有错误，敬请指正。

 

先随机产生一个[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]等价：

同时，还可以这样用。