使用嵌套列表推导创建随机0和1的矩阵

时间:2019-04-21 22:18:23

标签: python

我是Python的新手,我刚开始学习列表理解如何返回工作。我被困在一个棘手的运动中。

我想返回一个包含n列表的列表,这些列表包含n随机整数0或1。

from random import randint
def CreateGraph(n):
    return [R for i in range(n) R[L[i]=randint(0,1)]]
n=3
print(CreateGraph(n))

我想要的是这样的:

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

因为我的n是3。

相反,我得到了指向R[L[i]=randint(0,1)]的无效语法。

1 个答案:

答案 0 :(得分:1)

有什么帮助吗?

>>> n = 3
>>> n
3
>>> [[randint(0, 1) for _ in range(n)] for _ in range(n)]
[[0, 0, 1], [1, 0, 1], [1, 1, 0]]