如何在熊貓數據框中找到與另一列中的多個值相對應的列中具有值的所有行？

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