如何在具有相同 ID 的行上应用条件

Question

我有一个像这样的数据框：

df= [ ID  child_ID  STATUS1  
      123  11        OK    
      123  22        KO 
      123  23        OK 
      124  56        OK  
      124  45        OK

我想通过 ID 获得最终状态，如果其中一个 childid 是 KO，则意味着我的最终状态是 KO，所以我想要一个类似的数据框：

f= [      ID  Child_ID  STATUS1  Statusfinal
          123  11           OK       KO
          123  22           KO       KO
          123  23           OK       KO
          124  56           OK       OK
          124  45           OK       OK 

我该怎么做？

Answer 1

我会先计算一个额外的布尔列，然后使用 groupby 聚合：

resul = df.assign(FinalStatus = ((df['STATUS1'].str.lower() == 'ok') &
                  (df['STATUS2'].str.lower() == 'ok'))
          ).groupby('ID').agg({'STATUS1': 'first',
                   'STATUS2': 'first',
                   'FinalStatus': all}).reset_index()

在最后一列添加正确的标签，仅此而已：

resul['FinalStatus'] = np.where(resul['FinalStatus'], 'OK', 'KO')

给予：

   ID STATUS1 STATUS2 FinalStatus
0  12      OK      OK          KO
1  13      OK      OK          OK