python求度分布的程序

使用python求度分布的程序

# -*- coding: utf-8 -*-
"""
Created on Sun Jun 16 20:13:42 2019

@author: Administrator
"""

"""

这个函数的作用是将一个矩阵给转换成一个图，
矩阵以多维列表的形式存在，即列表的列表
此处的转换是针对无向图

根据邻接矩阵得到图之后，我们就可以调用networkx
里面的各种函数来分析图的性质，比如度分布，
平均路径程度，聚类系数等一系列图的拓扑性质

"""

import networkx as nx
import matplotlib.pyplot as plt


def degree_of_graph():
  
 # reload(sys) 
  G = nx.Graph()
  
  filestr = ""
  
  with open("matrix.txt") as files:
     for line in files:
         filestr += line.strip()
  
  #将字符串转换成列表
  matrix = eval(filestr)
  
  #存储y坐标的值
  y_value = []
  
  #存储x坐标的值
  x_value = []
  
  nodes = range(len(matrix))
  G.add_nodes_from(nodes)
 
  for i in range(len(matrix)):
    for j in range(len(matrix)):
      if(matrix[i][j] == 1):
		G.add_edge(i, j)
        
  print("每个节点的编号以及度值如下:")
  print(nx.degree(G)) 
  print("---------------------------------------------------------")
  print("每个度值的个数为:")
  print(nx.degree_histogram(G))
  print("---------------------------------------------------------")
  print("度值序列如下:")
  print(range(len(nx.degree_histogram(G))))
  print("---------------------------------------------------------")
  print("度值所占的比例如下:")
  for i in nx.degree_histogram(G):
      y_value.append(float(i)/sum(nx.degree_histogram(G)))
      
  for j in range(len(nx.degree_histogram(G))):
      x_value.append(j)
  print(y_value)
  print("---------------------------------------------------------")
  
  #使用x坐标和y坐标来构成一个字典
  x_y_value = {}
  
  for i in range(len(x_value)):
      for j in range(len(y_value)):
          if(i == j and y_value[j] != 0):
              x_y_value[x_value[i]] = y_value[j]
              
  
  print("去掉0以后的度值序列如下:")
  print(x_y_value.keys())
  print("去掉0以后的度值所占的比例如下:")
  print(x_y_value.values())
  fig = plt.figure()
  ax1 = fig.add_subplot(111)  

  #设置标题  
  #ax1.set_title("  ")  

  plt.xlabel('K')
  plt.ylabel("P(K)")

  plt.scatter(x_y_value.keys(),x_y_value.values(),c = "red",s=10)


degree_of_graph()