python - 如何 reshape 3d numpy 表？

Question

我有一个 shape=(2,3,4) 的 3d numpy 表，如下所示:

a = np.array([[[1., 2., 3., 4.],
        [1., 2., 3., 4.],
        [1., 2., 3., 4.]],

       [[5., 6., 7., 8.],
        [5., 6., 7., 8.],
        [5., 6., 7., 8.]]])

并希望以每个维度中的列堆叠成二维矩阵中的新列的方式 reshape 它。

1  5
1  5
1  5
2  6
2  6
2  6
3  7
3  7
3  7
4  8
4  8
4  8

Answer 1

给你:

res = a.T.reshape((-1,2))

输出:

array([[1., 5.],
       [1., 5.],
       [1., 5.],
       [2., 6.],
       [2., 6.],
       [2., 6.],
       [3., 7.],
       [3., 7.],
       [3., 7.],
       [4., 8.],
       [4., 8.],
       [4., 8.]])