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

前端之家收集整理的这篇文章主要介绍了python – 为pandas dataframe中的两列创建邻接矩阵前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
我有一个表格的数据框:
  1. index Name_A Name_B
  2. 0 Adam Ben
  3. 1 Chris David
  4. 2 Adam Chris
  5. 3 Ben Chris

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

  1. Adam Ben Chris David
  2. Adam 0 1 1 0
  3. Ben 0 0 1 0
  4. Chris 0 0 0 1
  5. David 0 0 0 0

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

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

解决方法

您可以在列和索引值的0700和 reindex之间使用 crosstabreindex
  1. df = pd.crosstab(df.Name_A,df.Name_B)
  2. print (df)
  3. Name_B Ben Chris David
  4. Name_A
  5. Adam 1 1 0
  6. Ben 0 1 0
  7. Chris 0 0 1
  8.  
  9. df = pd.crosstab(df.Name_A,df.Name_B)
  10. idx = df.columns.union(df.index)
  11. df = df.reindex(index = idx,columns=idx,fill_value=0)
  12. print (df)
  13. Adam Ben Chris David
  14. Adam 0 1 1 0
  15. Ben 0 0 1 0
  16. Chris 0 0 0 1
  17. David 0 0 0 0

猜你在找的Python相关文章