将二进制编码转换为类多标签python

Question

我在数据框中有这样的数据

Text	Label1	Label2	label3	label4	Labels
Hello my name is john	1	0	1	0

我想根据 1 和 0 来填充标签列，就像这样

Text	Label1	Label2	label3	label4	Labels
Hello my name is john	1	0	1	0	['Label1','Label3']

在python中我该怎么做？

Answer 1

好的，这是来自this answer 的未经测试的建议。尝试收集适当的列标签，然后在合适的函数上使用 DataFrame.apply：

test_cols = [c for c in df.columns if c[:5].lower() == 'label']
test_cols.remove('Labels')

def aggLabels(aSeries):
    return [lab for lab in test_cols if aSeries[lab]==1]

df['Labels'] = df.apply(aggLabels, axis=1)

正如我所说，这是未经测试的；可能需要对代码进行调整。