用途一:数据集太大,无法画图,取子图看看是有向图/无向图
import dgl
import matplotlib.pyplot as plt
import networkx as nx
G = dgl.DGLGraph()
G.add_nodes(5)
# G.add_edges([0, 1, 2, 3, 4], [1, 2, 3, 4, 0]) # 有向图
G.add_edges([0, 1, 2, 3, 4, 1, 2, 3, 4, 0], [1, 2, 3, 4, 0, 0, 1, 2, 3, 4]) # 无向图(双向图)
SG = G.subgraph([0, 1, 4])
SG.nodes()
# tensor([0, 1, 2])
SG.edges()
# (tensor([0, 2]), tensor([1, 0]))
SG.parent_nid
# tensor([0, 1, 4])
SG.parent_eid
# tensor([0, 4])nx_G1 = G.to_networkx()
nx_G2 = SG.to_networkx()
plt.subplot(121)
nx.draw_shell(nx_G1, with_labels=True, font_weight='bold') # 节点按序排列
plt.subplot(122)
nx.draw_shell(nx_G2, with_labels=True, font_weight='bold') # 节点按序排列
plt.show()
有向图的子图是有向图
无向图的子图是无向图