python - 如何使用Python或SQL获取数据框中一对项目的频率

Question

我的约会是这样的

d = {'name':['a','b','c','a','c','a','b','a','b','d'],'class_no.':[1,1,1,2,2,3,3,4,4,4]}
x = pd.DataFrame(data=d)
display(x)

我想知道两个人在同一个类(class)有多少次？例如，a 和 b 在 1、3 和 4 中一起上课。 a 和 c 在 1 和 2 中一起上课。所以 (a,b) =3, (a,c)=2。

有没有办法使用 pandas 或 SQL 来解决这个问题？谢谢

Answer 1

在 SQL 中，您可以自连接表并计算对数。注意:我限制对，以便第一个项目始终小于第二个项目，以避免重复结果((a,b) 和 (b,a)):

SELECT   a.name, b.name, COUNT(*)
FROM     mytable a
JOIN     mytable b ON a.class_no = b.class_no and a.name < b.bame
GROUP BY a.name, b.name