python - pandas - 检查数据框 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()