加入收藏 | 设为首页 | 会员中心 | 我要投稿 安卓应用网 (https://www.0791zz.com/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 编程开发 > Python > 正文

python – 为pandas dataframe中的两列创建邻接矩阵

发布时间:2020-05-27 18:59:27 所属栏目:Python 来源:互联网
导读:我有一个表格的数据框: index Name_A Name_B 0 Adam Ben 1 Chris David 2 Adam Chris 3 Ben Chris 我想获得Name_A和Name_B的邻接矩阵,即: Adam Ben Chris DavidAdam 0 1 1 0Ben 0

我有一个表格的数据框:

index  Name_A  Name_B
  0    Adam    Ben
  1    Chris   David
  2    Adam    Chris
  3    Ben     Chris

我想获得Name_A和Name_B的邻接矩阵,即:

Adam Ben Chris David
Adam   0    1    1     0
Ben    0    0    1     0
Chris  0    0    0     1
David  0    0    0     0

什么是他们最狡猾/可扩展的解决方法?

编辑:另外,我知道如果行Adam,Ben在数据集中,那么在其他一点,Ben,Adam也将在数据集中.

解决方法

您可以在列和索引值的0700和 reindex之间使用 crosstabreindex
df = pd.crosstab(df.Name_A,df.Name_B)
print (df)
Name_B  Ben  Chris  David
Name_A                   
Adam      1      1      0
Ben       0      1      0
Chris     0      0      1

df = pd.crosstab(df.Name_A,df.Name_B)
idx = df.columns.union(df.index)
df = df.reindex(index = idx,columns=idx,fill_value=0)
print (df)
       Adam  Ben  Chris  David
Adam      0    1      1      0
Ben       0    0      1      0
Chris     0    0      0      1
David     0    0      0      0

(编辑:安卓应用网)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读