使用Networkx,如何将graphml和其他格式写入字符串而不是文件?

时间:2012-06-15 08:35:44

标签: python networkx graphml

我非常浅薄地使用networkx。将图形写入文件(如graphml)很容易,但如何在不打扰文件系统的情况下将其保存为字符串?

Its doc说这是可能的。

2 个答案:

答案 0 :(得分:8)

大多数格式也有“生成器”。所以你可以这样做而不使用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>

答案 1 :(得分:5)

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