python - 使用 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))

这段代码按预期工作。然而，如您所见，它相当丑陋(而且很可能比应有的速度慢)。

我觉得，应该有一种更快和/或更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