我有 2 个节点 A 和 B 的列表,用于 networkx 中的图 G。
我想绘制节点,使列表 A 保留在左侧,列表 B 中的节点保留在右侧。
为此,我想知道窗口的大小,并为列表 A 中的节点执行类似操作,在窗口左侧有弹簧布局,对于右侧的列表 B 也是如此。
有没有办法这样做....?
import matplotlib.pyplot as plt
import networkx as nx
g = nx.Graph()
ListA = [1]
ListB = [2]
for node in ListA:
g.add_node(node)
for node in ListB:
g.add_node(node)
pos=nx.spring_layout(g) // want to change the position of all the nodes such that ListA is on left and ListB is on right
print pos
nx.draw_networkx_nodes(g,pos,nodelist=ListA,node_color='r')
nx.draw_networkx_nodes(g,pos,nodelist=ListB,node_color='g')
nx.draw_networkx_labels(g,pos)
plt.show()