如何在pandas数据框中创建一个新列,该列包含值为1的列索引

时间:2020-09-06 00:02:57

标签: python pandas dataframe

train是我的数据帧,它包含32列和24000行。从特征编号8开始(标签),这些列中的每一列的值都只是二进制(0,1)。我想创建一个新列并向其追加一个列表,其中包含标签为1的列的索引。这是我尝试过的操作,但遇到错误。是否有一种方法可以在存在多个标签的情况下执行此操作。热编码,但用于多标签。

for val in train.values:
    feats=val[:8]
    labels=val[8:]
    target=[k for k,i in enumerate(labels) if i==1]
    print(target)
    train['new_column']=target

0 个答案:

没有答案