Numpy合并两个数组我可以像这样制作一个numpy数组吗？

Question

我正在尝试这样的两个数组。它与column_stack不同，所以我无法从文档或谷歌搜索中找到如何做到这一点。

我有数组a和b。如何从他们那里获得c？

a = [[1, 2],[3, 4]]
b = [[5 , 6]]

c = [[[1, 2],[5]],
     [3, 4],[6]]]  

我需要将这些值输入到theanets。

Answer 1

In [54]:
a = np.array([[1, 2],[3, 4]])
a
Out[54]:
array([[1, 2],
       [3, 4]])

In [55]:
b = np.array([[5 , 6]])
b
Out[55]:
array([[5, 6]])


In [96]:
c = [[a[n].tolist() , b[:,n].tolist()] for n in range(len(a))]
c
Out[96]:
[[[1, 2], [5]], [[3, 4], [6]]]