python - 如何在矩阵数组中生成随机 1 和 0？

Question

我有一个矩阵，它目前只填充了 1。我如何使它填充随机的 1 和 0？

matrix5x5 = [[1 for row in range (5)] for col in range (5)]

for row in matrix5x5:
    for item in row:
        print(item,end=" ")
    print()
print("")

输出：

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

我想要类似的东西：

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

我发现了一些关于使用的东西，random.randint(0,1)但我不知道如何更改我当前的代码以包含上述内容。

标签： python

Answer 1

random使用包（而不是numpy等效包）修改代码：

matrix5x5 = [[random.randint(0,1) for _ in range(5)] for _ in range(5)]

for row in matrix5x5: 
    for item in row:
        print(item,end=" ") 
    print()
print("")

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

但老实说，numpy让它更快更容易！