python - Python初始化随机值的多维numpy数组

Question

如果我想构建一个 3 维数组并且我可以写这样的东西（x 是 3 维数组）

for i in range(pN): 
    for j in range(C): 
        for k in range(K+1):
            X[i][j][k] = random.uniform(0,1) #random initialize

但是我怎样才能使这段代码更具可读性呢？（例如，不要使用 3 for 循环）

谢谢！

标签： pythonnumpy

Answer 1

只需使用 numpy 随机函数：

https://docs.scipy.org/doc/numpy-1.15.1/reference/generated/numpy.random.rand.html

它将生成您想要的形状的随机数组（如 3x2x3）

 np.random.rand(3,2,3)