我非常肤浅地使用networkx。将图形写入文件很容易,例如 graphml,但是如何在不打扰文件系统的情况下将其保存到字符串中呢?
它的文档说这是可能的。
大多数格式也有一个“生成器”。所以你可以在不使用 StringIO 的情况下这样做:
In [1]: import networkx as nx
In [2]: G=nx.path_graph(4)
In [3]: s='\n'.join(nx.generate_graphml(G))
In [4]: print s
<graphml xmlns="http://graphml.graphdrawing.org/xmlns" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://graphml.graphdrawing.org/xmlns http://graphml.graphdrawing.org/xmlns/1.0/graphml.xsd">
<key attr.name="name" attr.type="string" for="graph" id="d0" />
<graph edgedefault="undirected">
<data key="d0">path_graph(4)</data>
<node id="0" />
<node id="1" />
<node id="2" />
<node id="3" />
<edge source="0" target="1" />
<edge source="1" target="2" />
<edge source="2" target="3" />
</graph>
</graphml>
正如 larsmans 评论的那样,可以使用StringIO
:
import networkx as nx
import StringIO
import itertools
g = nx.Graph()
edges = itertools.combinations([1,2,3,4], 2)
g.add_edges_from(edges)
# File-like object
output = StringIO.StringIO()
nx.write_graphml(g, output)
# And here's your string
gstr = output.getvalue()
print gstr
output.close()