python - Pandas 在一列中搜索在另一列中具有不同值的重复行

Question

我有一个 Pandas 数据框 df，我想为它找到列 A 的值相同但列 的值相同的所有行B 不同，例如:

       | A | B
    ---|---|---
     0 | 2 | x 
     1 | 2 | y 

我知道我可以使用 pd.concat(g for _, g in df.groupby('A') if len(g) > 1) 来获取具有重复值的行 A，但是如何添加第二个约束？

Answer 1

考虑到这一点，在 groupby 上调用 unique 是有意义的:

In [213]:
df = pd.DataFrame({'A':2, 'B':list('xxyzz')})
df

Out[213]:
   A  B
0  2  x
1  2  x
2  2  y
3  2  z
4  2  z

In [229]:
df.groupby('A')['B'].apply(lambda x: x.unique()).reset_index()

Out[229]:
   A          B
0  2  [x, y, z]