DGL_子图

用途一：数据集太大，无法画图，取子图看看是有向图/无向图

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()

有向图的子图是有向图
无向图的子图是无向图