Numpy：多维数组的真值比较

Question

因此，我正在编写一个数独解算器，使用9x9数组作为网格，使用9x9x9数组作为其可能性。由于我使用的回溯算法，我必须检查数独是否仍然有效，又名：

如果有一个字段不包含数字，并且没有剩余的可能性，则返回False。到目前为止，我实现了以下几点：

    for j in range(9):
        for i in range(9):
            if puzzle[j,i] == 0 and (solving[j,i] == 0).sum() == 9:
                return False
    return True

例如，如果（j，i）处的正方形包含选项2、3和7，则相应的可能性数组为：

    solving[j,i] = array([0, 2, 3, 0, 0, 0, 7, 0, 0], dtype = int8)

我试图在求解器中避免这样的循环。我尝试如下：

    solving[where(puzzle == 0)]

这返回了一个nx9数组，其中n是我的拼图中的空方块数。如果形状为1x9的n个数组中的任何一个正好等于

    array([0,0,0,0,0,0,0,0,0], dtype = int8)

但是，我在包括. any（）和. all（）实验在内的各种尝试中没有任何成功。这是我使用的最新代码：

    def is_valid(self, puzzle, solving):
        if (solving[py.where(puzzle == 0)].all() == 0).sum() > 0:
            return False
        return True

... 它总是返回False。正确的编码方式是什么？

Answer 1

我认为以下是您要做的检查：

np.all(puzzle | np.any(solving, axis=-1)

如果所有单元格都分配了一个值（拼图），或者（|）在求解中至少有一个非零条目，则返回True（np.any（求解，轴=-1））。