热线电话:13121318867

登录
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.0000
2
关注作者
收藏
评论(0)

发表评论

暂无数据
推荐帖子