如何在熊猫数据框中找到与另一列中的多个值相对应的列中具有值的所有行？

Question

我需要找出 A 列中有多少值对应于 B 列中的多个值。

一个	乙
1	猫
2	狗
3	鼠
1	马
1	猫
2	鸟
3	鼠

我如何遍历 A 中的所有值并获得一个告诉我的输出：

1 有猫和马

2 有狗和鸟

Answer 1

如果您只想知道哪些值符合您的条件：

df.loc[~df.duplicated(keep=False), 'A'].unique()

输出： array([2, 1])

要获取值：

out = df.groupby('A')['B'].agg(set)
out = out[out.str.len()>1]

输出：

A
1    {cat, horse}
2     {bird, dog}
Name: B, dtype: object