Networkx: Преобразование мультиграфа в простой граф с взвешенными краями

У меня есть объект multigraph и хотел бы преобразовать его в простой объект графа с взвешенными ребрами. Я просмотрел документацию по сети и не могу найти встроенную функцию для достижения этой цели. Мне просто интересно, знал ли кто-нибудь о встроенной функции в networkx, которая могла бы достичь этой цели. Я посмотрел на функции to_directed(), to_undirected(), но они не служат моей цели.

Ответ 1

Вот один из способов создания взвешенного графика из взвешенного мультиграфа путем суммирования весов:

import networkx as nx
# weighted MultiGraph
M = nx.MultiGraph()
M.add_edge(1,2,weight=7)
M.add_edge(1,2,weight=19)
M.add_edge(2,3,weight=42)

# create weighted graph from M
G = nx.Graph()
for u,v,data in M.edges(data=True):
    w = data['weight'] if 'weight' in data else 1.0
    if G.has_edge(u,v):
        G[u][v]['weight'] += w
    else:
        G.add_edge(u, v, weight=w)

print(G.edges(data=True))
# [(1, 2, {'weight': 26}), (2, 3, {'weight': 42})]

Ответ 2

Один очень простой способ сделать это - просто передать ваш мультиграф в качестве входного сигнала в Graph.

import networkx as nx

G = nx.MultiGraph()
G.add_nodes_from([1,2,3])
G.add_edges_from([(1, 2), (1, 2), (1, 3), (2, 3), (2, 3)])

G2 = nx.Graph(G)

Это создаст неориентированный график вашего мультиграфа, где несколько ребер будут объединены в отдельные ребра. Однако, если у вас есть разные атрибуты для ребер, которые объединяются, я не знаю, есть ли способ определить, какой атрибут сохранен.