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

时间:2020-10-11 15:55:33

标签: python numpy

如果我要构建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用于循环)

谢谢!

4 个答案:

答案 0 :(得分:2)

只需使用numpy随机函数:

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

它将生成所需形状的随机数组(如3x2x3)

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

答案 1 :(得分:1)

您可以使用numpy

import numpy as np
np.random.random((pN,C,K))

答案 2 :(得分:1)

请改用numpy。

首先使用

安装
pip install -U numpy

在终端或Windows命令提示符下。

然后在python程序中将其导入:

import numpy as np

然后使用random.rand

np.random.rand((pN,C,K))

答案 3 :(得分:1)

import numpy as np

X = np.random.uniform(0, 1, (x, y, z))
# x, y, z would represent the size of each dimension

From the numpy documentation