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



如果我想构建一个三维数组我可以写这样的东西(x是三维阵列(

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用于循环(

谢谢!

只需使用numpy随机函数:

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

它将生成你想要的形状的随机数组(如3x2x3(

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

您可以使用numpy

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

请改用numpy。

首先安装

pip install -U numpy

在终端或窗口命令提示符下。

然后在python程序中导入:

import numpy as np

然后使用random.rand:

np.random.rand((pN,C,K))
import numpy as np
X = np.random.uniform(0, 1, (x, y, z))
# x, y, z would represent the size of each dimension

从numpy文档

最新更新