Python 如何水平堆叠数组

发布于09月21日

我有一个数组:

> a
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], 
        [25, 26, 27, 28], 
        [29, 30, 31, 32]]])

我怎样才能把它reshape 成这样:

[[1, 2, 3, 4, 17, 18, 19, 20],
 [5, 6, 7, 8, 21, 22, 23, 24],
 [9, 10, 11, 12, 25, 26, 27, 28],
 [13, 14, 15, 16, 29, 30, 31, 32]]

我try 了reshape函数，但它返回以下内容:

[[1, 2, 3, 4, 5, 6, 7, 8],
 [9, 10, 11, 12, 13, 14, 15, 16],
 [17, 18, 19, 20, 21, 22, 23, 24],
 [25, 26, 27, 28, 29, 30, 31, 32]]

有什么解决办法吗？

非常感谢

推荐答案

Numpy将此功能内置在方法hSTACK中

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], 
        [25, 26, 27, 28], 
        [29, 30, 31, 32]]])

np.hstack(a)

# array([[ 1,  2,  3,  4, 17, 18, 19, 20],
#        [ 5,  6,  7,  8, 21, 22, 23, 24],
#        [ 9, 10, 11, 12, 25, 26, 27, 28],
#        [13, 14, 15, 16, 29, 30, 31, 32]])