python - 常见项目选择 - Pandas Dataframe

Question

好吧，我知道这应该很容易，但我找不到解决方案。我正在用 python 进行一些社交网络分析，并有一个包含这样的数据的 pandas Dataframe(连接)...

uid  |   name   |  fuID  |  friendName   | 
1    |  Bob     |   f1   |   Jimmy       |
1    |  Bob     |   f2   |   Frank       |
1    |  Bob     |   f3   |   Sal         |
2    |  Mark    |   f4   |   Leon        |
2    |  Mark    |   f1   |   Jimmy       |
2    |  Mark    |   f5   |   Artie       |
3    |  Steve   |   f1   |   Jimmy       |
3    |  Steve   |   f5   |   Artie       |
3    |  Steve   |   f6   |   Samuel      |

目标是找到所有共享的 friend ，换句话说，我希望能够找到所有具有多个连接的friendName/fuID以及这些连接是什么。例如，吉米与鲍勃、马克和史蒂夫有联系。

我可以很容易地创建一个 groupby ...connections.groupby("uid") 并假设有一种方法可以与组进行比较以找到共同的项目，但我遇到了困难。我开始构建一个复杂的 for 循环来迭代各组，然后就停止了，因为我知道必须有一种简单的方法来做到这一点，而我只是缺少这种方法。

任何帮助将不胜感激。

Answer 1

好吧，你就快到了——

fdfg = connections.groupby('friendName')
for k, v in fdfg:
    if len(v) > 1:
        print k
        print v.name

Artie
5     Mark
7    Steve
Name: name, dtype: object
Jimmy
0      Bob
4     Mark
6    Steve
Name: name, dtype: object

可能有一种更简洁的方法来做到这一点。