python - 在python中创建一个随机矩阵

Question

我正在尝试用 numpy 创建一个 nxn 随机数的随机方阵。当然，我能够生成足够的随机数，但我在使用 numpy 创建可变长度矩阵时遇到了问题。这是我到目前为止所得到的:

def testMatrix(size):
    a = []
    for i in range(0, size*size):
        a.append(randint(0, 5))

如何将这个列表放入大小为 x 大小的数组中？

Answer 1

尝试

np.random.randint(0, 5, size=(s, s))