三维维数numpy

时间:2019-01-22 08:49:36

标签: python numpy stack repeat

我有一个矩阵W,我想重复并在第三轴上堆叠x次。这样做最简单的方法是什么?我可以在两个代码行中做到这一点,想知道是否可以在一个代码行中做到这一点:

np.repeat(W,x,axis=1)
W=W.reshape(W.shap[0],W.shap[1],x)

1 个答案:

答案 0 :(得分:1)

您可以添加一个新的空轴(使用np。newaxis),然后沿该轴重复。

W = np.repeat(W[:, :, np.newaxis], x, axis=2)