连接np.arrays python

Question

如果出现以下情况，我该怎么办

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

b = np.array([0,1])

我搜索将a和b连接起来，结果将是：

np.array([1,2,3,0],[5,6,7,1])

非常感谢

Answer 1

问题是将a作为列向量水平连接b。

<concat>( |1 2 3|, |0| )
          |5 6 7|  |1|

可以使用np.hstack完成合并，并且可以通过添加新轴将converted this blog转换为列向量：

b

Answer 2

使用numpy广播的concatenate

np.concatenate([a,b[:,None]],1)
Out[1053]: 
array([[1, 2, 3, 0],
       [5, 6, 7, 1]])

Answer 3

执行此操作的更多方法是避免广播，并使用为此设计的功能：numpy.column_stack：

np.column_stack([a, b])

array([[1, 2, 3, 0],
       [5, 6, 7, 1]])