如何在 python 中将元组表示为二维数组？

Question

想象一个 NxN 棋盘，我有一个元组t = (0,3,2,1) ，它代表每一列的棋子位置（col = 索引），每个数字代表行，从底部的 0 开始。

对于这个例子，它有 4 列，第一块在 row=0（底行），第二块在 row=3（第四/最高行），第三块在 row=2（从底部数第三行），第四块一块是从底部开始的第二行。

我想将其表示为二维数组，如下所示：

[[0,1,0,0],
 [0,0,1,0],
 [0,0,0,1],
 [1,0,0,0]]

我能够使用此代码生成二维数组

pieces_locations = (0,3,2,1)
pieces_locations = list(pieces_locations)

table_size = len(pieces_locations)

arr = [[0 for col in range(table_size)] for row in range(table_size)]

但是，我无法将 1 分配到正确的位置。

我能够理解这一点：arr[row][col]，但是行是倒置的（0 是顶部，N 是底部）。

Answer 1

首先创建二维零列表。

arr = [[0] * table_size for _ in range(table_size)]

然后遍历位置，用1替换适当的元素。

for col, row in enumerate(pieces_location, 1):
    arr[-row][col] = 1

Answer 2

在你制作列表后使用它（0 的矩阵）** 如果位置列表不与行数一样长，程序将崩溃（使用 try 和 except 来反击）

for x, i in enumerate(range(1, len(arr))):
    arr[-i][pieces_locations[x]] = 1

这应该会为您提供所需的 output，希望对您有所帮助

Answer 3

我能够弄清楚，虽然我确信有一个移动方便的方法。

pieces_locations = (0,3,2,1)
pieces_locations = list(pieces_locations)

table_size = len(pieces_locations)

arr = [[0 for col in range(table_size)] for row in range(table_size)]


for row in range(0, table_size):
        arr[row][pieces_locations.index(row)] = 1


res = arr[::-1]
print (res)