如何在数据帧的开头插入列向量? [重复]

问题描述 投票:0回答:1

我有以下两个变量:

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

python pandas numpy
1个回答
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)
© www.soinside.com 2019 - 2024. All rights reserved.