沿第一轴连接两个numpy数组

Question

我有一个名为(1538,4)的维度为X_scaled的数组，另一个名为(1538,1)的维度为Y_mlp的数组。我想将Y_mlp添加到X_scaled，以便Y_mlp成为X_scaled中的第五列。我怎么能这样做？

Answer 1

您正在寻找np.hstack。

numpy.hstack(tup) 

     

获取一系列数组并将它们水平堆叠以形成一个数组   阵列。

import numpy as np

X_scaled, Y_mlp = ..., ...

Y_mlp = Y_mlp.reshape(-1, 1)   
out = np.hstack((X_scaled, Y_mlp))

print(out.shape)

输出：

(5, 5)

沿第二维发生连接。