如何更快地遍历具有2个维度的Python numpy.ndarray

所以，我只是想让它更快：

for x in range(matrix.shape[0]):
        for y in range(matrix.shape[1]):
            if matrix[x][y] == 2 or matrix[x][y] == 3 or matrix[x][y] == 4 or matrix[x][y] == 5 or matrix[x][y] == 6:
                if x not in heights:
                    heights.append(x)

只需迭代2x2矩阵（通常是18x18或22x22的整数），然后检查它是否为x。但是它有点慢，我想知道这是最快的方法。

非常感谢你！