0

我对 Networkx 很陌生。我正在尝试导入由random_layout()函数生成的布局位置。我不知道该怎么做。

生成布局位置的代码：

G = nx.random_geometric_graph(10, 0.5)
pos = nx.random_layout(G)
nx.set_node_attributes(G, 'pos', pos)
f = open("graphLayout.txt", 'wb')
f.write("%s" % pos)
f.close()
print pos
filename = "ipRandomGrid.txt"
fh = open(filename, 'wb')
nx.write_adjlist(G, fh)
#nx.write_graphml(G, sys.stdout)
nx.draw(G)
plt.show()
fh.close()

文件：ipRandomGrid.txt

# GMT Tue Dec 06 04:28:27 2011
# Random Geometric Graph
0 1 3 4 6 8 9 
1 3 4 6 8 9 
2 4 7 
3 8 6 
4 5 6 7 8 
5 8 9 6 7 
6 7 8 9 
7 9 
8 9 
9

我将节点adjlist和布局都存储在文件中。现在我想生成具有相同布局和adjlist其他文件的图形。我尝试使用以下代码生成它。谁能帮我弄清楚这里出了什么问题。

导入时的代码：伪代码

G = nx.Graph() 
G = nx.read_adjlist("ipRandomGrid.txt")
# load POS value from file 
nx.draw(G)
nx.draw_networkx_nodes(G, pos, nodelist=['1','2'], node_color='b')
plt.show()
4

1 回答 1

0

该nx.random_layout函数返回一个字典映射节点到位置。作为pos一个 Python 对象，您不想像在f.write("%s" % pos). 这为您提供了一个包含您的字典的文件，但读回它并不容易。

相反，pos使用为该任务设计的标准库模块之一进行序列化，例如，json或pickle. 它们的接口基本相同，所以我将仅展示如何使用pickle. 存储是：

with open("graphLayout.txt", 'wb') as f:
    pickle.dump(pos, f)

重载是：

with open("graphLayout.txt", 'rb') as f:
    pos = pickle.load(f)
于 2011-12-06T10:39:04.340 回答