pandas - 檢查dataframe groupby中的非唯一值

Question

我有這個簡單的數據幀df ：

a,b
1,2
1,3
1,4
1,2
2,1
2,2
2,3
2,5
2,5

我想，以檢查是否有一式兩份b與在相對於每個組a 。 到目前為止，我做了以下事情：

g = df.groupby('a')['b'].unique()

返回：

a
1       [2, 3, 4]
2    [1, 2, 3, 5]

但我想有一個列表，每個組中的a ，與多次出現在b 。 在這種情況下的預期輸出將是：

a
1    [2]
2    [5]

Answer 1

g=df.groupby('a')['b'].value_counts()
g.where(g>1).dropna()

Answer 2

我們可以使用duplicated

print(df[df.duplicated()].drop_duplicates())