2019-02-14
阅读量:
718
根据条件进行数据帧过滤
我有一个像数据框架
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
解决办法:
可以使用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






评论(0)


暂无数据
推荐帖子
0条评论
0条评论
0条评论