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的,这是个坑