python - 如何按索引连接数组的数组？

Question

如果我有两个或多个 n 维数组，每个数组还包含数组，我如何按索引连接所有数组？

一个例子是:

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)