代码之家  ›  专栏  ›  技术社区  ›  SkyWalker

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

  •  0
  • SkyWalker  · 技术社区  · 6 年前

    我有以下两个变量:

    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 回复  |  直到 6 年前
        1
  •  1
  •   jezrael    6 年前

    使用 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)