從數據文件構造圖

Question

我有一個文本文件，其數據如下： AD 15 BA 11 CH 2 . . . . . . AD 15 BA 11 CH 2 . . . . . .

我在Python中使用Dataframe讀取數據。 然后我要創建一個在第1列和第2列中具有頂點的圖形，第3列是權重。

如何從數據創建圖形？ 謝謝！

Answer 1

這是一個使用networkx的簡單示例：

import networkx as nx

node_list = list(set(list(df['col1']) + list(df['col2'])))  
data = [tuple(x) for x in df.values.tolist()]
# [('A', 'D', 15), ('B', 'A', 11), ('C', 'H', 2), . . .]

G = nx.Graph()
G.add_nodes_from(node_list)
G.add_weighted_edges_from(data)