如何创建随机矩阵？

Question

我想创建一个随机的矩阵，比如 [[[100, 50, 25], [22, 75, 195]]] 这样的结构。

我的代码是：

n = 1
r = 2
e = 3

sup = []

for i in range(n):
    sup1 = []
    for c in range(r):
       sup0 = list (random.randint (200, 0, e))
       sup1.append (sup0)
    sup.append (sup1)    

print sup

但是 Python 给我报错了。

Answer 1

这个应该可以工作（我不知道 e 是干嘛用的）：

sup = [[random.randint(0, 200) for _ in range(r)] for _ in range(n)]

Answer 2

你可以使用 numpy 这个工具，直接生成一个你想要大小的随机矩阵，并且这些数值可以在你指定的范围内。

>>> numpy.random.randint(low = 0, high = 200, size=(2, 4))
array([[ 75,  21, 132,  90],
       [112,  11, 104, 114]])

>>> r = 2
>>> n = 1
>>> numpy.random.randint(low = 0, high = 200, size=(r, n))
array([[94],
       [51]])

更多细节请查看这里