如何通过图形工具使用 CSV 文件数据在 Python 中创建图形？

Question

我正在尝试使用 csv 文件中的图形工具 (https://graph-tool.skewed.de) 创建一个图形，其内容如下：

A,B,50
A,C,34
C,D,55
D,D,80
A,D,90
B,D,78

现在我想创建一个图，其中 A、B、C、D 作为节点，第三列编号作为边。我正在使用图形工具库。第三列数字显示了A、B和A、C等共有的项目。

我可以通过“networkx”（read_edgelist 等）来实现，但我想用图形工具来实现。

Answer 1

您可以使用 add_edge_list() 来添加边列表。如果它们存储的名称与自动分配的索引不同，它将返回一个包含列表中名称的字符串列表。

示例：

from graph_tool.all import *
import csv

g=Graph(directed=False)
csv_E = csv.reader(open('*text_input*'))

e_weight=g.new_edge_property('float')
v_names=g.add_edge_list(csv_E,hashed=True,string_vals=True,eprops=[e_weight])  
#this will assign the weight to the propery map *e_weight* and the names to *v_names*

graph_draw(g, vertex_text=v_names)

Answer 2

假设您已经知道如何在 Python 中读取 CSV 文件（例如，使用CSV library），网站上的文档explain how to do this very clearly.

类似

import graph_tool
g = Graph(directed=False)


# this is the result of csv.parse(file)
list_of_edges = [['A', 'B', 50], ['A','C',34], ['C','D',55], ['D','D',80], ['A','D',90], ['B','D',78]]

vertices = {}

for e in list_of_edges:
    if e[0] not in vertices:
        vertices[e[0]] = True
    if e[1] not in vertices:
        vertices[e[1]] = True


for d in vertices:
    vertices[d] = g.add_vertex()

for edge in list_of_edges:
    g.add_edge(vertices[edge[0]], vertices[edge[1]])