python - 如何从字典中向 igraph 对象添加边

Question

我正在使用字典来表示一组边。我使用字典的键来表示边缘，使用值来表示权重。字典目前如下所示:

{(0, 1): 2, (1, 2): 6, (0, 2): 3}

我试试这个:

edges, weights = [], []
for edge, weight in dict_edges.items():
    edges += [edge]
    weights.append(weight)

g.add_edges(edges)
g.es["weight"] = weights

但是，我不知道是否有更快或更简洁的方法来做到这一点。

有人对如何改进我的新功能有任何建议吗？

Answer 1

你所做的一切都很好；也许 for 循环可以替换为 zip 调用。如果您使用的是 Python 2.x::

from itertools import izip
edges, weights = izip(*dict_edges.iteritems())
g = Graph(edges, edge_attrs={"weight": weights})

如果您使用的是 Python 3.x::

edges, weights = zip(*dict_edges.items())
g = Graph(edges, edge_attrs={"weight": weights})