从数据文件构造图

时间:2017-04-10 11:35:50

标签: python dataframe graph

我有一个包含以下数据的文本文件: A D 15 B A 11 C H 2 . . . . . .

我在Python中使用Dataframe读取数据。然后我想在第1列和第1列中创建一个带顶点的图形。 2和第3列是重量。

如何根据数据创建图表? 谢谢!

1 个答案:

答案 0 :(得分:1)

以下是使用networkx的简短示例:

import networkx as nx

node_list = list(set(list(df['col1']) + list(df['col2']))
data = [tuple(x) for x 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)