连接 np.arrays python答案

【问题标题】：Concatenate np.arrays python连接 np.arrays python
【发布时间】：2019-03-11 04:15:21
【问题描述】：

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

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])

非常感谢

【问题讨论】：

【解决方案1】：

更numpythonic的方法是避免广播，并使用为此设计的函数：numpy.column_stack:

np.column_stack([a, b])

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

【讨论】：

【解决方案2】：

使用numpy 广播和concatenate

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

【讨论】：

【解决方案3】：

问题是水平连接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]])

【讨论】：