python - 如何使用 Numpy 生成随机 N 维？

Question

我目前正在尝试制作 N 维矩阵。

import numpy as np
fitness_landscape = np.random.uniform (0, 1, size = (N, 2., 2., 2., 2.))

在这段代码中，我想减少递归二维。例如，像下面的数组(虽然它不起作用):

import numpy as np
K = 4
fitness_landscape = np.random.uniform (0, 1, size = (N, 2. * K))

有什么方法可以满足我的需求吗？

谢谢。

Answer 1

In [9]: K = 5

In [10]: fitness_landscape = np.random.uniform (0, 1, size = (N,) + (2,) * K)

In [11]: fitness_landscape.shape
Out[11]: (3, 2, 2, 2, 2, 2)