numpy：将数组沿着新轴延伸？

Question

当然，必须有办法做到这一点......我无法解决这个问题。我有一个（9,4）数组，我想沿着第三轴重复4096次...所以它变成了简单的（9,4,4096），每个数值都来自9,4数组。简单地沿新轴线重复4096次。

如果我半信半疑的3D图是有道理的（对角线为Z轴）

4| /off to 4096 
3|/
2|/
1|/_ _ _ _ _ _ _ _ _ 
    1 2 3 4 5 6 7 8 9 

干杯

编辑：只是为了澄清，这里强调的是（9,4）阵列对新轴的4096行中的每一行重复执行。设想一个截面 - 每个原始（9,4）阵列是4096长方体之一。

Answer 1

这里有一种方法：

import scipy 
X = scipy.rand(9,4,1) 
Y = X.repeat(4096,2) 

如果X就交给你了，因为只有（9,4），然后

import scipy 
X = scipy.rand(9,4) 
Y = X.reshape(9,4,1).repeat(4096,2) 

Answer 2

您也可以依靠广播规则重复填充重新排列的阵列：

import numpy 
X = numpy.random.rand(9,4) 
Y = numpy.resize(X,(4096,9,4)) 

如果您不喜欢用这种方式订购的轴，则可以转置：

Z = Y.transpose(1,2,0) 

Answer 3

问题是超级老了，但这里的另一种选择呢：

import numpy as np 
X = np.random.rand(9,4) 
Y = np.dstack([X] * 4096)