使用 igraph 从字典填充图形

Question

我有一本字典，里面有关注者-被关注者的信息。例如：

a --> b,c,d
b --> c,d
c --> d

这里的意思是，a在b,c,d之后。 b 跟随 c,d。 c 跟随 d。我有以下对应的字典：

{'a': ['b', 'c', 'd'], 'b': ['c', 'd'], 'c': ['d']}

我想根据 igraph 中的字典创建一个有向图。

我目前正在做的是：

import igraph

d = {'a': ['b', 'c', 'd'], 'b': ['c', 'd'], 'c': ['d']}

edges = []
vertices_set = set()
for key in d:
    vertices_set.add( key )
    for item in d[key]:
        edges.append( (key, item) )
        vertices_set.add( item )

# vertices_set: set(['a', 'c', 'b', 'd'])
# edges: [('a', 'b'), ('a', 'c'), ('a', 'd'), ('c', 'd'), ('b', 'c'), ('b', 'd')]

g = igraph.Graph( directed=True )
g.add_vertices( len(vertices_set) )
g.vs['name'] = list(vertices_set)
g.add_edges( edges )

print(len(g.vs))
print(len(g.es))

并且此代码按预期工作。然而，如您所见，它相当丑陋（而且很可能比它应该的慢）。

我觉得应该有一种更快 and/or 更 pythonic 的方式来做到这一点。有什么意见吗？

Answer 1

您可以使用Graph.TupleList函数：

>>> from igraph import Graph
>>> d = {'a': ['b', 'c', 'd'], 'b': ['c', 'd'], 'c': ['d']}
>>> g = Graph.TupleList([(k, v) for k, vs in d.iteritems() for v in vs])
>>> print g
IGRAPH UN-- 4 6 --
+ attr: name (v)
+ edges (vertex names):
a -- b, c, d   b -- a, c, d   c -- a, b, d   d -- a, b, c

使用 igraph 从字典填充图形

Populating a graph from a dictionary using igraph

python

igraph