python - 根据条件过滤数据帧

Question

我有一个像

这样的数据框

colA  colB  colC
 A      B    C 
 A      D    C 
 B      B    E 
 A      D    C 
 C      B    C 

我想按这样的优先级过滤它们:如果 colC == E 则返回 E，之后检查 colB == D 返回 D 否则返回 colA输出是

colA  colB  colC   final
 A      B    C      A 
 A      D    C      D
 B      B    E      E 
 A      D    C      D
 C      B    C      C

Answer 1

你可以使用 np.select ，它允许您根据条件列表在多个值中进行选择:

m1 = df.colC =='E'
m2 = df.colB =='D'
df.loc[:,'final'] = np.select([m1,m2], ['E', 'D'], default=df.colA)

   colA colB colC final
0    A    B    C     A
1    A    D    C     D
2    B    B    E     E
3    A    D    C     D
4    C    B    C     C