如何在数据框的开头插入列向量？ [复制]

Question

我有以下两个变量：

print('Column vector type %s and shape %s' % (type(target), target[0:X_train.shape[0]].shape))
print('Data frame type %s and shape %s' % (type(X_train), X_train.shape))

这个输出：

Column vector type <class 'numpy.ndarray'> and shape (87145,)
Data frame type <class 'pandas.core.frame.DataFrame'> and shape (87145, 11)

我想插入 target 列向量作为此帧/矩阵的第一列...我该怎么做？

我的最终目标是能够使用corr 函数计算附加到预测变量或设计矩阵的响应变量的相关矩阵。

Answer 1

使用DataFrame.insert:

X_train.insert(0,'target',target) 
print (X_train)

对于分配相同长度的数组，如DataFrame:

X_train.insert(0,'target',target[:X_train.shape[0]]) 
print (X_train)