时间戳

首页 > 解决方案 > 使用python在单位矩阵中填充几个0

python - 使用python在单位矩阵中填充几个0

问题描述

我想用几个 0 填充一个单位矩阵。我有X,Y坐标[(0, 0), (0, 1), (0, 2), (1, 2), (1, 3), (2, 3)]，那些坐标应该是 0，其他的应该用 1 填充。

这是针对最短路径问题的。该算法将输出最短路径作为坐标

[(0, 0), (0, 1), (0, 2), (1, 2), (1, 3), (2, 3)]

我怎样才能在python中做到这一点？

样本输入矩阵[4][4]

1 1 1 1
1 1 1 1
1 1 1 1
1 1 1 1

样本输出矩阵[4][4]

0 1 1 1
0 1 1 1
0 0 1 1
1 0 0 1

标签： pythonalgorithmmatrix

解决方案

只需遍历您在坐标数组中的索引并将要在这些索引处修改的数组的值分配为 0，如下所示：

for i in coordinatesArray:
    toModify[i[0]][i[1]] = 0

如果要修改的数组尚未填充，请先使用 for 循环添加它们：

for i in range(x):
    toModify.append(y * [1])

推荐阅读