如何使用Numpy制作隨機N維？

Question

我正在嘗試制作N維矩陣。

import numpy as np
fitness_landscape = np.random.uniform (0, 1, size = (N, 2., 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)