我正在使用Metis for Python
,这是Metis的Python包装器(一种图形分区软件)。我已经安装了所有东西,并且似乎可以正常工作,但是我不明白如何构造图来输入。
以下是一个在线示例:http://metis.readthedocs.org/en/latest/#example
>>> import networkx as nx
>>> import metis
>>> G = metis.example_networkx()
>>> (edgecuts, parts) = metis.part_graph(G, 3)
>>> colors = ['red','blue','green']
>>> for i, p in enumerate(parts):
... G.node[i]['color'] = colors[p]
...
>>> nx.write_dot(G, 'example.dot') # Requires pydot or pygraphviz
我运行了这个示例,它运行良好。但是,在此示例中,它们从未指定如何构造图“ example_networkx()”。
我试图通过networkx构造图:http://metis.readthedocs.org/en/latest/#metis.networkx_to_metis
我的代码是:
>>> A=nx.Graph()
>>> A.add_edges_from([(3,1),(2,3),(1,2),(3,4),(4,5),(5,6),(5,7),(7,6),(4,10),(10,8),(10,9),(8,9)])
>>> G = metis.networkx_to_metis(A)
>>> (edgecuts, parts) = metis.part_graph(G, 3)
我在最后一行出现错误。该错误可追溯到Metis内置代码中的以下几行:
in part_graph(graph, nparts, tpwgts, ubvec, recursive, **opts)
graph = adjlist_to_metis(graph, nodew, nodesz)
in adjlist_to_metis(adjlist, nodew, nodesz)
m2 = sum(map(len, adjlist))
TypeError: object of type 'c_long' has no len()
我也尝试过通过邻接表构造图:http://metis.readthedocs.org/en/latest/#metis.adjlist_to_metis
但这会产生与以前相同的错误。
我想知道是否有人遇到了这个问题,或者知道我在做什么错。
我在Centos 6.5上使用python 2.7
最佳答案
metis.part_graph接受图的networkx和邻接表表示。
当您构建networkx图时,您几乎是正确的。但是,您应该直接将此图传递给part_graph函数,而不是首先将其转换为metis对象,因为part_graph函数不直接接受metis类型图。
给定numpy中的邻接矩阵A
,示例可以是:
# since weights should all be integers
G = networkx.from_numpy_matrix(np.int32(A))
# otherwise metis will not recognize you have a weighted graph
G.graph['edge_weight_attr']='weight'
[cost, parts] = metis.part_graph(G, nparts=30, recursive=True)