如果我有两个或多个 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]]])
最佳答案
请注意,由于内部列表的长度不同,因此您最终构建了 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)
关于python - 如何按索引连接数组的数组?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/30470494/