我有一本字典,其中包含关注者-关注者信息。例如:
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的方式来做到这一点。有意见吗?
最佳答案
您可以使用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
关于python - 使用 igraph 从字典填充图形,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/34571534/