如果我想构建一个三维数组我可以写这样的东西(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文档