python - 连接 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 来完成连接和 b 可以是 converted通过添加新轴将其转换为列向量:

>>> np.hstack([a, b[:, np.newaxis]])
array([[1, 2, 3, 0],
       [5, 6, 7, 1]])