Numpy 3D数组矢量化实现与原生循环代码计算结果不匹配问题排查

核心错误：索引对应关系错位 #

你的矢量化判断逻辑本身没有问题，但和循环代码中的赋值位置存在错位：

• 矢量化代码arr[2:, :, :] == arr[:-2, :, :]得到的isArrEqual形状为(14,73,144)，其中第0维下标m对应的是循环中k = m + 1的判断结果：
  • m=0对应k=1的判断条件arr[2,j,i] == arr[0,j,i]
  • m=13对应k=14的判断条件arr[15,j,i] == arr[13,j,i]
• 你给出的循环代码存在明显的索引笔误：将k=1_{14的判断结果赋值给`boolArray[k,j,i]`，而`boolArray`第0维长度为14，合法下标仅为0}13，k=14时会直接触发数组越界。如果你实际运行的循环代码没有报错，说明你实际用的赋值逻辑应为boolArray[k-1,j,i] = True，该逻辑才和isArrEqual的结果完全匹配。如果直接把isArrEqual当作最终的boolArray，或者循环的赋值逻辑写错，就会出现结果不匹配的情况。

次要排查点：浮点数精度问题 #

如果你的arr存储的是浮点数类型，直接用==判断相等会因为浮点计算的精度误差，导致本该相等的值被判定为不相等。这种情况下可以改用近似相等判断：

isArrEqual = np.isclose(arr[2:,:,:], arr[0:-2,:,:])

内容的提问来源于stack exchange，提问作者gansub