python - 使用 Pandas 计算关系表中的共同关注者

Question

我有一个像这样的 pandas DataFrame:

   from_user  to_user
0        123      456
1        894      135
2        179      890
3        456      123 

每行包含两个 ID，反射(reflect) from_user 是否“跟随”to_user。如何使用 pandas 计算 DataFrame 中相互关注者的总数？

在上面的例子中，答案应该是 1(用户 123 和 456)。

Answer 1

一种方法是使用 MultiIndex 集合操作:

In [11]: i1 = df.set_index(["from_user", "to_user"]).index

In [12]: i2 = df.set_index(["to_user", "from_user"]).index

In [13]: (i1 & i2).levels[0]
Out[13]: Int64Index([123, 456], dtype='int64')

要获得计数，您必须将此索引的长度除以 2:

In [14]: len(i1 & i2) // 2
Out[14]: 1