将数据从 2d 转换为 3d numpy 数组

Question

如果我有一个像这样的二维数组：

array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

我想得到一个像这样的 3d 数组：

array([[[0, 4, 8],
        [1, 5, 9]],

       [[2, 6, 10],
        [3, 7, 11]]])

我应该如何重塑数组以获得我想要的？

Answer 1

-

In [11]: a  # Input array
Out[11]: 
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

In [12]: a.reshape(-1,2,2).transpose(1,2,0)
Out[12]: 
array([[[ 0,  4,  8],
        [ 1,  5,  9]],

       [[ 2,  6, 10],
        [ 3,  7, 11]]])

与np.moveaxis-

np.moveaxis(a.reshape(-1,2,2), 0,-1)

概括它并假设您希望沿第一个轴的长度为 no 的一半。列数 -

In [16]: m,n = a.shape

In [17]: a.reshape(m,-1,2).transpose(1,2,0)
Out[17]: 
array([[[ 0,  4,  8],
        [ 1,  5,  9]],

       [[ 2,  6, 10],
        [ 3,  7, 11]]])

如果该长度应该是 2 -

In [15]: a.reshape(m,2,-1).transpose(1,2,0)
Out[15]: 
array([[[ 0,  4,  8],
        [ 1,  5,  9]],

       [[ 2,  6, 10],
        [ 3,  7, 11]]])

将数据从 2d 转换为 3d numpy 数组

convert data from 2d to 3d numpy array

numpy

reshape

python-2.7