networkx库
pip install --upgrade networkx
点和边示例:
import networkx as nx
import matplotlib.pyplot as plt
G = nx.Graph() #初始化一个图
G.add_node('a')
G.add_node('b')
G.add_node('c')
G.add_node('d')
G.add_node('e')
G.add_edge('a','b') #连接a、b得到ab边
G.add_edge('a','d')
G.add_edge('a','e')
G.add_edge('a','c')
nx.draw(G,with_labels=True)
plt.show()
规则图:
import networkx as nx
import matplotlib.pyplot as plt
RG = nx.random_graphs.random_regular_graph(3,20) #生成包含20个节点、每个节点有3个邻居的规则图RG
pos = nx.spectral_layout(RG) #定义一个布局,此处采用了spectral布局方式,后变还会介绍其它布局方式,注意图形上的区别
nx.draw(RG,pos,with_labels=False,node_size = 30) #绘制规则图的图形,with_labels决定节点是非带标签(编号),node_size是节点的直径
plt.show() #显示图形
无向图示例:
import networkx as nx
import matplotlib.pyplot as plt
# BA scale-free degree network
# generalize BA network which has 20 nodes, m = 1
BA = nx.random_graphs.barabasi_albert_graph(20, 1)
# spring layout
pos = nx.spring_layout(BA)
nx.draw(BA, pos, with_labels = False, node_size = 30)
plt.show()
# 导入相关依赖
from matplotlib import pyplot as plt
import networkx as nx
import numpy as np
# 生成随机数据
G = nx.erdos_renyi_graph(50,0.5)
# 指定画布大小
plt.figure(figsize=(18,18))
# 生成新的图
G_new = nx.Graph()
# 依据图中边的数量,生成同样长度的随机权重值
weightList = {}
for i in range(len(G.edges())+1):
weightList[i] = np.random.rand()
# 将生成的随机权重复制给G_new图
i = 0
for edge in G.edges():
i += 1
G_new.add_edges_from([(edge[0], edge[1], {'weight': weightList[i]})])
# 绘制G_new图
nx.draw_networkx(G_new)
plt.show()