numpy通过堆叠列将2d矩阵重塑为3d矩阵

Question

我有一个形状为(x,y)的numpy矩阵。 我想得到一个张量，它是形状为(x,1)的y向量的垂直堆栈。 可以说我有以下矩阵：

array([[ 1.,  2.],
       [ 1.,  2.],
       [ 1.,  2.]])

当我做np.reshape(2,3,1) 。 我会得到：

array([[[ 1.],
        [ 2.],
        [ 1.]],
       [[ 2.],
        [ 1.],
        [ 2.]]])

但我想要这个：

array([[[ 1.],
        [ 1.],
        [ 1.]],
       [[ 2.],
        [ 2.],
        [ 2.]]])

Answer 1

In [135]: arr=np.array([[1,2],[1,2],[1,2]])
In [136]: arr.shape
Out[136]: (3, 2)
In [137]: arr.transpose()
Out[137]: 
array([[1, 1, 1],
       [2, 2, 2]])
In [138]: arr.transpose()[:,:,None]
Out[138]: 
array([[[1],
        [1],
        [1]],

       [[2],
        [2],
        [2]]])

您想要一个形状（2,3,1）。 从（3,2）开始，这意味着您必须切换2个轴，并添加一个。 可以按任何顺序完成。 在这里，我选择转置进行切换，并选择[:,:,None]添加尺寸。

arr.reshape(3,2,1).swapaxes(0,1)也可以使用。 更晦涩的是np.stack(arr[:,:,None],1) 。