如何在networkx可视化中为不同的节点集提供不同的标签选项？

import networkx as nx
import matplotlib.pyplot as plt

G = nx.Graph()

G.add_node(1)
G.add_node(2)
G.add_edge(1, 2)

G.nodes[1]['label'] = 'Node 1'
G.nodes[2]['label'] = 'Node 2'

node_set1 = [1]
node_set2 = [2]

labels = {}
for node in node_set1:
    labels[node] = G.nodes[node]['label'] + ' (Set 1)'
for node in node_set2:
    labels[node] = G.nodes[node]['label'] + ' (Set 2)'

pos = nx.spring_layout(G)  # 设置节点布局
nx.draw(G, pos, with_labels=True, node_color='lightblue')  # 绘制图形
nx.draw_networkx_labels(G, pos, labels, font_color='red')  # 设置节点标签
plt.show()  # 显示图形