pandas - 如何通过检查标志查找值

Question

数据框在下方

uid,col1,col2,flag
1001,a,b,{'a':True,'b':False}
1002,a,b,{'a':False,'b':True}

出

a
b

通过检查标志，如果a是 true，则在 a列上打印 out，如果 b标志是 true，则在 b列上打印 out

Answer 1

IIUC，可以在 dot 构造函数之后使用 DataFrame :

m=pd.DataFrame(df['flag'].tolist()).fillna(False)
final=df.assign(New=m.dot(m.columns))
print(final)

    uid col1 col2         flag New
0  1001    a    b  {'a': True}   a
1  1002    a    b  {'b': True}   b