将 Numpy 数组合并为二维数组

Question

我有 2 个像这样的 numpy 数组：

    a = [[a,b,c],
         [d,e,f]]
    b = [[g,h,i],
        [k,l,m]]

我想将它们合并到另一个 numpy 数组中，如下所示：

c = [[[a,g],[b,h],[c,i]],
    [[d,k],[e,l],[f,m]]]

我该怎么做？

Answer 1

可以使用dstack函数，即

a = np.array([[1,2,3],
     [4,5,6]])
print(a)
b = np.array([[10,20,30],
    [40,50,60]])
print(b)

c = np.dstack((a,b))
print(c)

这将输出

[[1 2 3]
 [4 5 6]]
[[10 20 30]
 [40 50 60]]
[[[ 1 10]
  [ 2 20]
  [ 3 30]]

 [[ 4 40]
  [ 5 50]
  [ 6 60]]]

Merging Numpy Arrays into 2D array