Python-在数据帧中查找相同ID的值

我有如下数据框，如下所示

ID    TYPE    POLICY_NUMBER       DISB_AMT
738   20      FLDINC MSH39990     1
738   21      MSH39990            3848
750   20      INF395737           1
750   21      INF395737 FLDINCL   2350
892   20      SJK389743           3904
892   21      MSH284989           1     

我试图按ID分组并提取保单编号，然后在其他类型中搜索，例如：（TYPE=20或21）如果两种类型中的保单编号相同，则检查两行中的DISB_AMT>1。如果为true，则不要将其附加到数据帧

例如：ID738在两行中具有相同的保单编号MSH39990。我写了一个脚本，只提取数字，以便比较。ID 738具有相同的保单编号。现在，我们检查DISB_AMT是否大于1。在第一行中，它不是>1。在第二行，我们有3848>1。不要在结果中包含此ID。对于ID 892，由于两种类型中的保单编号不相同，我们仅检查类型21的DISB_AMT>1。因为它不是>1，所以我们将此行添加到结果数据框中

如何将其与其他类型进行比较，检查策略编号是否相同，并构建其余逻辑

预期产出

ID    TYPE    POLICY_NUMBER       DISB_AMT
892   21      MSH284989           1

代码

data = [{"ID":738,"TYPE":20,"POLICY_NUMBER":"FLDINC MSH39990","DISB_AMT":1},
        {"ID":738,"TYPE":21,"POLICY_NUMBER":"MSH39990","DISB_AMT":3848},
        {"ID":750,"TYPE":20,"POLICY_NUMBER":"INF395737","DISB_AMT":1},
        {"ID":750,"TYPE":21,"POLICY_NUMBER":"INF395737 FLDINCL","DISB_AMT":2350},
        {"ID":892,"TYPE":20,"POLICY_NUMBER":"SJK389743","DISB_AMT":3904},
        {"ID":892,"TYPE":21,"POLICY_NUMBER":"MSH284989","DISB_AMT":1}
        ]

df=pd.DataFrame(data)

df['CLEANED_POL_NBR']=df.POLICY_NUMBER.str.extract('(\d+)')

IIUC：

输出：

   DISB_AMT   ID POLICY_NUMBER  TYPE CLEANED_POL_NBR
5         1  892     MSH284989    21          284989

---

如果保单编号匹配，我们将检查这两种类型（20或21）的解散金额。如果保单编号不匹配，我们将检查类型是否为21，如果解散金额为1。当策略不匹配时，如何添加此附加筛选器，即TYPE=21？

   DISB_AMT   ID POLICY_NUMBER  TYPE CLEANED_POL_NBR
5         1  892     MSH284989    21          284989