如何从pandas两列创建共现矩阵？
如何从pandas两列创建共现矩阵？
提问于 2021-01-19 02:32:48
我有一个很小的数据集，例如：
import pandas as pd
df = pd.DataFrame({'a': [1,2,3,4,5,6,7,8,9,10], 'b': [11,22,11,22,33,11,22,44,11,22]})
df
我想找出列a的b列的值的共现情况。
我尝试过的：
df_co = pd.get_dummies(a.a).groupby(a.b).apply(max)
df_co但这不是一个共现矩阵。所以我也试了一下：
df_co.T.dot(df_co)这给了我：
这是计算共生矩阵的正确方法吗？
回答 1
Stack Overflow用户
发布于 2021-01-19 08:00:08
您可以使用带有虚拟列的df.pivot来表示count=1
df.assign(v=1).pivot('a','b').fillna(0) v
b 11 22 33 44
a
1 1.0 0.0 0.0 0.0
2 0.0 1.0 0.0 0.0
3 1.0 0.0 0.0 0.0
4 0.0 1.0 0.0 0.0
5 0.0 0.0 1.0 0.0
6 1.0 0.0 0.0 0.0
7 0.0 1.0 0.0 0.0
8 0.0 0.0 0.0 1.0
9 1.0 0.0 0.0 0.0
10 0.0 1.0 0.0 0.0或者，就像@Quang Hoang建议的那样，试试pd.crosstab
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/65780181
复制相关文章
相似问题
腾讯云开发者
