我有一个包含以下数据的文本文件:
A D 15
B A 11
C H 2
. . .
. . .
我在Python中使用Dataframe读取数据。然后我想在第1列和第1列中创建一个带顶点的图形。 2和第3列是重量。
如何根据数据创建图表? 谢谢!
答案 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)