将两个数组的每一行连接成一个数组

时间:2014-04-09 16:27:19

标签: python numpy

假设我有两个数组a和b。

a.shape is (95, 300)
b.shape is (95, 3)

如何通过连接95行中的每一行来获得新的数组c?

c.shape is (95, 303)

1 个答案:

答案 0 :(得分:3)

IIUC,您可以使用hstack

>>> a = np.ones((95, 300))
>>> b = np.ones((95, 3)) * 2
>>> a.shape
(95, 300)
>>> b.shape
(95, 3)
>>> c = np.hstack((a,b))
>>> c
array([[ 1.,  1.,  1., ...,  2.,  2.,  2.],
       [ 1.,  1.,  1., ...,  2.,  2.,  2.],
       [ 1.,  1.,  1., ...,  2.,  2.,  2.],
       ..., 
       [ 1.,  1.,  1., ...,  2.,  2.,  2.],
       [ 1.,  1.,  1., ...,  2.,  2.,  2.],
       [ 1.,  1.,  1., ...,  2.,  2.,  2.]])
>>> c.shape
(95, 303)