python 中 numpy 数组的引索踩坑之旅

python 中 numpy 数组的引索踩坑之旅

我今天在做图像数组按像素点周围取小区域处理算法，在使用numpy里的数组引索切片时发现一个超级不显眼的坑。我搞了好久才排除bug

例如：图像灰度数据A(5*5)
A[2:]

import numpy as np 
a = np.zeros((5,5,3))
b1 = a[2:,2:,0]
b2 = a[:2,:2,0]
print(b1,'\n', b1.shape, '\n' , b2, '\n',b2.shape )
>>>[[0. 0. 0.]
 [0. 0. 0.]
 [0. 0. 0.]] 
 (3, 3) 
 [[0. 0.]
 [0. 0.]] 
 (2, 2)

着重看b1,与b2的维大小，两个是差1的，这是个坑