如果第三列不同，pandas dataframe刪除2列上的重復項

Question

我想使用熊貓解決以下問題。 我想刪除具有相同id和y值且具有不同z列的行。 這是一個簡單的示例：

  id  y  z
0  a  0  3
1  a  0  1
2  a  0  3
3  b  3  4
4  b  0  2
5  b  0  2

我想獲得以下結果

  id  y  z
0  b  3  4
1  b  0  2
2  b  0  2

有人有主意嗎？

Answer 1

聯合會

x = df.set_index(['id', 'y'])
x.loc[x.groupby(level=[0,1]).transform(lambda k: k.duplicated(keep=False).all() or k.size==1).z].reset_index()

    id  y   z
0   b   3   4
1   b   0   2
2   b   0   2