将1d和2d numpy数组合并为一个元组

Question

我这里有两个numpy数组：

a = np.array([1, 2]) 
b = np.array([(1, 2, 3), (4, 5, 6)])

我想将它们合并到tuple中，以便a的元素是垂直的，如下所示：

[(1, [1, 2, 3]), (2, [4, 5, 6])]

您能否提供有关如何执行此操作的建议？

Answer 1

zip符合条款吗？

>>> a = np.array([1,2])
>>> b = np.array([(1,2,3),(4,5,6)])
>>> zip(a,b)
[(1, array([1, 2, 3])), (2, array([4, 5, 6]))]