python - 水平连接numpy向量和矩阵

Question

我有以下 numpy 向量 m 和矩阵 n

import numpy as np
m = np.array([360., 130., 1.])
n = np.array([[60., 90., 120.], 
              [30., 120., 90.],
              [1.,  1.,   1. ]])

我想做的是将它们水平连接起来

np.array([[60., 90., 120.,360.], 
          [30., 120., 90., 130.],
          [1.,  1.,   1., 1. ]])

有什么方法可以做到？

我试过了但是失败了:

np.concatenate(n,m.T,axis=1)

Answer 1

>>> np.hstack((n,np.array([m]).T))
array([[  60.,   90.,  120.,  360.],
       [  30.,  120.,   90.,  130.],
       [   1.,    1.,    1.,    1.]])

问题是因为 m 只有一维，它的转置仍然是一样的。在进行转置之前，您需要使其具有形状 (1,3) 而不是 (3,)。

按照 DSM 在评论中的建议，np.hstack((n,m[:,None])) 是一个更好的方法。