python - 检查列是否在列表中，如果没有则删除并将值添加到新列

Question

我有一个这样的数据库:

df = pd.DataFrame( { 
        "A" : [0,0,1,1,0,1] , 
        "B" : [1,0,0,1,1,0],
        "C" : [0,0,0,1,1,0],
        "D" : [1,1,0,0,0,1]} )

看起来像这样:

    A   B   C   D
0   0   1   0   1
1   0   0   0   1
2   1   0   0   0
3   1   1   1   0
4   0   1   1   0
5   1   0   0   1

我有一个希望保留的列列表allowed_columns = ["A","B"]。这意味着我们摆脱了C 和D。但是，当删除列时，如果有值 1，我想在新列中lost 中注明。这就是我想要实现的目标:

    A   B   lost    
0   0   1   1   
1   0   0   1   
2   1   0   0   
3   1   1   1   
4   0   1   1   
5   1   0   1   

为了简化问题，我们可以假设C和D不能同时具有值​​1。我怎样才能实现这个目标？

Answer 1

取允许列的子集，然后使用 df.columns.difference 取您删除的所有内容的最大值

df = (df[allowed_columns]
       .assign(lost=df[df.columns.difference(allowed_columns)].max(1)))