python - 如何在 python 中创建 3x3 数独 block 列表

Question

我需要帮助为数独中的 9 个 3x3 block 创建一个列表。所以我有一个代表原始数独板的列表列表(零表示空):

board=[[2,0,0,0,0,0,0,6,0],
       [0,0,0,0,7,5,0,3,0],
       [0,4,8,0,9,0,1,0,0],
       [0,0,0,3,0,0,0,0,0],
       [3,0,0,0,1,0,0,0,9],
       [0,0,0,0,0,8,0,0,0],
       [0,0,1,0,2,0,5,7,0],
       [0,8,0,7,3,0,0,0,0],
       [0,9,0,0,0,0,0,0,4]]

我需要将这些变成包含 3x3 block 的列表列表。例如:

[[2,0,0,0,0,0,0,4,8],[etc]]

我尝试创建一个名为“ block ”的列表，其中包含 9 个其他列表，每个列表中只有零。所以它看起来像:

blocks=[[0,0,0,0,0,0,0,0,0],[etc]

然后我使用 while 循环来更改列表中的值:

BLOCK_COUNT=0
BOARD_COUNT=0
while BLOCK_COUNT<len(blocks):
    blocks[BLOCK_COUNT][0]=board[BOARD_COUNT][BOARD_COUNT]
    blocks[BLOCK_COUNT][1]=board[BOARD_COUNT][BOARD_COUNT+1]
    blocks[BLOCK_COUNT][2]=board[BOARD_COUNT][BOARD_COUNT+2]
    blocks[BLOCK_COUNT][3]=board[BOARD_COUNT+1][BOARD_COUNT]
    blocks[BLOCK_COUNT][4]=board[BOARD_COUNT+1][BOARD_COUNT+1]
    blocks[BLOCK_COUNT][5]=board[BOARD_COUNT+1][BOARD_COUNT+2]
    blocks[BLOCK_COUNT][6]=board[BOARD_COUNT+2][BOARD_COUNT]
    blocks[BLOCK_COUNT][7]=board[BOARD_COUNT+2][BOARD_COUNT+1]
    blocks[BLOCK_COUNT][8]=board[BOARD_COUNT+2][BOARD_COUNT+2]
    BLOCK_COUNT+=1
    BOARD_COUNT+=3

然而，这给了我一个索引错误。如果我创建 2 个 while 循环，其中“BLOCK_COUNT”分别为 3 和 6，那么我会得到更好的答案，但对于某些人来说，它仍然没有给我正确的 3x3 block 。所以我对如何做到这一点几乎不知所措。谢谢。

Answer 1

def getBlocks(board):
    answer = []
    for r,c in itertools.product(range(3), repeat=2):
        answer.append([board[r+i][c+j] for i,j in itertools.product(range(0, 9, 3), repeat=2)])
    return answer

当然，你可以用一个列表理解来代替整个事情:

answer = [[board[r+i][c+j] for i,j in itertools.product(range(0, 9, 3), repeat=2)]
          for r,c in itertools.product(range(3), repeat=2)]

如果您对不使用任何内置函数来完成繁重工作的版本感兴趣:

def getBlocks(board):
    answer = []
    for r in range(3):
        for c in range(3):
            block = []
            for i in range(3):
                for j in range(3):
                    block.append(board[3*r + i][3*c + j])
            answer.append(block)
    return answer

那么这里发生了什么？:

好吧，首先，我们决定迭代我们想要的 9 个 block 。这些由 r 和 c 变量控制。这也是为什么我们在访问板上的数字时将它们乘以 3(因为每个 block 都是边长为 3 的正方形)。

接下来，我们要遍历每个 block 中的元素。翻译:查找每个 3x3 block 内的数字。 block 中每个元素的索引由 i 和 j 控制。所以我们有 i 和 j 来控制我们想要访问的元素，还有 r 和 c，它们是它们与电路板本身的偏移量，决定了我们想要的“ block ”的位置。现在我们开始比赛了。

对于每个 r 和 c(注意每个都在 range(3) 上循环，所以有 9 个 (r, c) 对——我们之后的 9 个 block )，遍历 block 中的 9 个元素(9 个 (i,j) 对)。现在，只需根据 (r,c) 偏移量的相对位置访问元素(3*r 给出相关 block 的第一行，并添加 i 给出所需元素的行。类似地，3*c 给出相关 block 的第一列，添加 j 给出相关 block 的列必需的元素。因此，我们有我们想要的元素的坐标)。现在，我们将元素添加到 block。

一旦我们遍历了 block 中的所有元素，我们就将 block 本身添加到答案中，很快!我们完成了