如何生成随机值的 N 维数组？

Question

目前，我正在写这个：

    x, y = np.array([random.randint(0, 9), random.randint(0, 9), random.randint(0, 9)]), \
           np.array([random.randint(0, 9), random.randint(0, 9), random.randint(0, 9)])

是否有内置的 NumPy 函数可以让我做得更好？

我说 N 维数组不是因为我想用随机值生成一个随机大小的向量，而是为了其他人可以将它应用于他们自己的向量。

Answer 1

您可以使用：

import numpy as np
np.random.randint(0, 9, shape)

例如，使用shape=(2, 3) ：

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