Python中的共现矩阵仅给出两列

Question

抱歉这个简单的问题，但是网上的参考资料似乎都与 NLP 有关，我认为我的场景相当简单。

我有一个类似的类别数据框：

df = pd.DataFrame({'x': [1,2,62,3,3], 'y':[5,6,6,1,1]})

我想创建一个方阵，将 x 和 u 中的所有唯一值作为列/行。在这种情况下，它将是 [1,2,3,5,6,62]，矩阵中的值是每对类别出现的频率。所以第 1 行和第 5 列的值 = 1，第 3 行、第 1 列的值 = 2。

在此先感谢

Answer 1

df.pivot_table(index='x', columns='y', aggfunc='size').fillna(0)

您可以使用pivot_table 来表示聚合函数是每个组合的计数（size），行（索引）是x，列是y（当然你也可以这样做反之亦然）。

我为不存在的组合添加了fillna(0)，这样看起来会更好。