Python numpy将3d列表重塑为2d数组

Question

我想改信

编辑:下面的代码是列表的3D列表，

[
    [ 
        [
            [1,2,3,],
            [4,5,6,],
        ],
        [
            [7,8,9,],
            [10,11,12,], 
        ],
    ],
    [ 
        [
            [A,B,C,],
            [D,E,F,], 
        ],
        [
            [G,H,I,],
            [J,K,L,], 
        ],
    ],
]

转到

[

[1,2,3],
[4,5,6],
[7,8,9],
[10,11,12],
[A,B,C],
[D,E,F],
[G,H,I],
[J,K,L]

]

我试过numpy.flatten，但没有成功https://docs.scipy.org/doc/numpy/reference/generated/numpy.ndarray.flatten.html

Answer 1

重塑是你的工具。

下面是一个自包含的示例：

import numpy as np
a = np.array([
    [[[1,2,3] , [4,5,6]],
     [[7,8,9] , [10,11,12]]],
    [[[13,14,15] , [16,17,18]],
     [[19,20,21] , [22,23,24]]]
    ])

a.shape
>>> (2, 2, 2, 3)

a.reshape(8,3)
>>> array([[ 1,  2,  3],
>>>        [ 4,  5,  6],
>>>        [ 7,  8,  9],
>>>        [10, 11, 12],
>>>        [13, 14, 15],
>>>        [16, 17, 18],
>>>        [19, 20, 21],
>>>        [22, 23, 24]])