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

Posted 2023-03-12

【中文标题】如何在数据框的开头插入列向量？ [复制]

【英文标题】：How to insert column vector at the beginning of a data frame? [duplicate]

【发布时间】：2019-06-28 00:06:57

【问题描述】：

我有以下两个变量：

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 函数计算附加到预测变量或设计矩阵的响应变量的相关矩阵。

【参考方案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)