如何合并2个numpy数组？

Question

我觉得我缺少了一些文档，但是在这个特定的例子中我找不到任何东西—所有的东西都是关于连接或堆叠数组的。在

我有数组x和数组y都是形状（2，3）

x=[[1,2,3]，[4,5,6]]

y=[[7,8,9]，[10,11,12]]

x = 1 2 3
    4 5 6

y = 7 8 9
    10 11 12

我想要一个形状为（2，3，2）的数组z，它看起来像这样

z=[[1,7]，[2,8]，[3,9]]，[[4,10]，[5,11]，[6,12]]]

^{pr2}$

基本上在每个位置连接x和y元素。在

Answer 1

听起来你要找的函数是stack()，用它在第三维度上堆叠。在

import numpy as np

x = np.asarray([[1,2,3],[4,5,6]])
y = np.asarray([[7,8,9],[10,11,12]])
z = np.stack((x, y), 2)

Answer 2

如果有二维矩阵，可以使用numpy.dstack()：

z = np.dstack((x, y))

Answer 3

In [39]: z = np.concatenate((x[...,None], y[...,None]), axis=2)

In [40]: z
Out[40]: 
array([[[ 1,  7],
        [ 2,  8],
        [ 3,  9]],

       [[ 4, 10],
        [ 5, 11],
        [ 6, 12]]])