在同一个第一维度上合并多个numpy数组



我有两个第一维度相同的numpy数组ab,例如a.shape -> (3, 100)b.shape -> (3, 20, 20)

我需要将它们(尽可能有效地(转换为具有相同第一维度(此处为3(的数组(或嵌套列表或dict?(ab。因此,当我调用ab[1]时,它应该返回类似[a[1], b[1]]的内容(或者用不同的说法ab[:][0] == aab[:][1] == b(。

我觉得这是一个容易的问题,但我不知道该找什么,而且我已经在这个问题上坐了太久了。

没有一种方法可以创建具有不同形状数组的组合NumPy数组。您可以使用一个列表并将它们放在同一行中。

ab = [[a[i],b[i]] for i in range(3)]
(ab[0][0] == a[0]).all()
>>>True

最新更新