沿第一轴连接两个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) 

级联沿着第二维度发生。