比较Numpy阵列

Question

因此，我有两个大小相等的2D numpy数组，都是在两个不同的曲面上使用pygame.surfarray.array3d方法获得的。

数组中的每个值也是[a， b， c]形式的数组（所以我基本上有一个带有1D元素的2D数组）。

基本上，我想根据条件对两者进行比较：

if any(val1 != val2) and all(val1 != [0, 0, 0]):
# can't be equal and val1 cant be [0, 0, 0] 

有没有更有效的方法可以做到这一点，而无需简单地迭代任何一个数组，如下所示？

for y in range(len(array1)):
    for x in range(len(array1[y])):
        val1 = array1[y,x]; val2 = array[y,x]
        if any(val1 != val2) and all(val1 != [0, 0, 0]):
            # do something

Answer 1

import numpy as np
if np.any(array1 != array2) and not np.any(np.all(a == 0, axis=-1))

np.any（array1！=array2）正在比较“大”3D数组的每个元素。然而，这相当于将每个x和y的val1与val2进行比较。

另一个条件，np.any（np.all（a==0，axis=-1））稍微复杂一点。最里面的np.all（a==0，轴=-1）创建一个布尔值的二维数组。每个值都设置为True或False，具体取决于最后一个维度中的所有值是否为0。外部条件检查2D数组中的任何值是否为True，这意味着array1[y，x]中有一个元素等于[0，0，0]。