熊猫-根据特定条件查找数据框中是否存在值

时间:2019-12-08 06:39:16

标签: pandas

我有一个数据框,其中包含票证列表以及其冲刺和状态,如下所示:

ticket,sprint,status
101,sprint_1,Closed
102,sprint_1,Open
103,sprint_2,Working
103,sprint_3,Fixed
103,sprint_4,Fixed
103,sprint_5,Open
103,sprint_6,Closed

我正在尝试查找特定冲刺中不是Closed的票,如果它们是其中的另一个冲刺。

在给定的样本集中,我们看到故障单102在给定的冲刺中未完成,但是没有将来的冲刺,这是故障单103sprint_2移至{{ 1}},并在sprint_3中关闭。

如果给定冲刺中没有sprint_3的查找票证,如果它们有另一个条目用于以后的冲刺,我会尝试添加它们。

预期输出:

Closed

1 个答案:

答案 0 :(得分:1)

使用:

#test equal
m1 = df['status'].eq('Open')
#test all duplicated tickets
m2 = df['ticket'].duplicated(keep=False)
#test all duplicated sprints
m3 = df['sprint'].duplicated(keep=False)
#test equal
m4 = df['status'].eq('Closed')
#test if at least one Open per group
m5 = m1.groupby(df['ticket']).transform('any')

df['part_of_future_sprint_if_not_closed'] = (m2 & ~m4 & m5).astype(int)
df['no_future_sprint_planned_open_tickets'] = (m1 & ~m2 & m3).astype(int)

print (df)
   ticket    sprint   status  part_of_future_sprint_if_not_closed  \
0     101  sprint_1   Closed                                    0   
1     102  sprint_1     Open                                    0   
2     103  sprint_2  Working                                    1   
3     103  sprint_3    Fixed                                    1   
4     103  sprint_4    Fixed                                    1   
5     103  sprint_5     Open                                    1   
6     103  sprint_6   Closed                                    0   

   no_future_sprint_planned_open_tickets  
0                                      0  
1                                      1  
2                                      0  
3                                      0  
4                                      0  
5                                      0  
6                                      0