我有两个第一维度相同的numpy数组a
和b
,例如a.shape -> (3, 100)
和b.shape -> (3, 20, 20)
。
我需要将它们(尽可能有效地(转换为具有相同第一维度(此处为3
(的数组(或嵌套列表或dict?(ab
。因此,当我调用ab[1]
时,它应该返回类似[a[1], b[1]]
的内容(或者用不同的说法ab[:][0] == a
和ab[:][1] == b
(。
我觉得这是一个容易的问题,但我不知道该找什么,而且我已经在这个问题上坐了太久了。
没有一种方法可以创建具有不同形状数组的组合NumPy数组。您可以使用一个列表并将它们放在同一行中。
ab = [[a[i],b[i]] for i in range(3)]
(ab[0][0] == a[0]).all()
>>>True