如何通过索引连接数组的数组？

Question

如果我有两个或更多个n-dimenisonal数组，每个数组都包含数组，那么如何通过索引连接所有数组？

一个例子是：

A = np.array([[[1,2], [3,4]], [[4,5,6], [1]]])
B = np.array([[[10,20], [3,9]], [[7,5], [2]]])

我想得到

C = np.array([[[1,2,10,20], [3,4,3,9]], [[4,5,6,7,5], [1,2]]])

Answer 1

请注意，由于您的内部列表长度不同，因此您最终构建了2×2个列表数组：

>>> A.shape
(2, 2)

但是在numpy中，+以元素方式行事，而在列表中，+会进行连接。所以：

>>> C = A + B
>>> C
array([[[1, 2, 10, 20], [3, 4, 3, 9]],
       [[4, 5, 6, 7, 5], [1, 2]]], dtype=object)