python - 根据行出现选择数据框结果

Question

给定以下 DataFrame，我如何才能仅检索 IS_TESTED 具有 True 和 False 值的值。

d = pd.DataFrame({"ID":[700,700,701,702,702,703],"IS_TESTED":[True,False,True,False,True,True],"TEST_NAME":["A","B","A","A","B","A"]})

在下面的例子中，我想要的结果应该是:

700 True  A 
700 False B 
702 False A 
702 True  B

因为 701 和 703 只出现一次。

Answer 1

使用 groupby 和 nunique

d[d.groupby('ID').IS_TESTED.transform('nunique') > 1]


    ID.         IS_TESTED   TEST_NAME
0   700         True         A
1   700         False        B
3   702         False        A
4   702         True         B