如何在numpy中创建三维矩阵，如matlab中的a(:,:,:)

Question

如何在numpy中创建一个三维矩阵，就像在matlab中用a(:,:,:)那样。我尝试把在matlab中创建三维矩阵的代码转换成python，但我不知道怎么在numpy中创建三维矩阵或数组。

Answer 1

a=np.empty((2,3,5))

这个代码创建了一个2x3x5的数组。（如果你想让数组里的值初始化为0，可以使用np.zeros。）

你也可以改变已经存在的数组的形状：

a=np.arange(30).reshape(2,3,5)

np.arange(30) 会创建一个一维数组，里面的值从0到29。reshape()方法会返回一个包含相同数据但形状不同的数组。