附加两个只有一个维匹配的数组

时间:2015-07-30 21:16:41

标签: python numpy

我有两个看起来像这样的numpy数组

a1 = array([p1,p2,p3])
a2 = array([1,2,3],[4,5,6],[7,8,9])

我希望将它们附加到它们看起来像这样:

array([p1,1,2,3],[p2,4,5,6],[p3,7,8,9])

我尝试了各种np.append,.concatenate,.insert,但无济于事。大多数时候我得到一个错误,上面写着“所有输入数组必须具有相同的维数”。有人知道这个伎俩吗?

1 个答案:

答案 0 :(得分:2)

使用numpy.insert

>>> a1 = np.array([10, 20, 30])
>>> a2 = np.array([[1,2,3],[4,5,6],[7,8,9]])
>>> np.insert(a2, 0, a1, axis=1)
array([[10,  1,  2,  3],
       [20,  4,  5,  6],
       [30,  7,  8,  9]])