如何使用python-igraph从指定结构的TXT文件绘制图

我来帮你一步步搞定这个需求，核心难点是处理带空格的Category列，然后正确构建节点与边的关联关系：

1. 先安装python-igraph #

如果还没安装依赖包，先执行这条命令：

pip install python-igraph

2. 读取并解析TXT文件 #

你的Category列包含空格（比如Gadgets & Games），直接用split()会把它拆成多个字段，所以得用更精准的方式拆分每行：取第一个元素作为Id，最后两个元素作为Related Id 1和Related Id 2，中间的所有元素拼接成完整的Category。

示例代码如下：

from igraph import Graph, plot

# 初始化图对象，无向图用False，有向图改成True
g = Graph(directed=False)

# 存储所有节点（自动去重）和边
nodes = set()
edges = []

# 读取目标TXT文件
with open('your_data.txt', 'r') as f:
    for line in f:
        line = line.strip()
        if not line:
            continue  # 跳过空行
        
        parts = line.split()
        # 拆分字段：第一个是节点Id，最后两个是关联节点，中间拼接为Category
        node_id = parts[0]
        related_node_1 = parts[-2]
        related_node_2 = parts[-1]
        category = ' '.join(parts[1:-2])  # 可选：如果需要保留Category信息
        
        # 把所有节点加入集合（自动去重）
        nodes.update([node_id, related_node_1, related_node_2])
        # 添加两条关联边：Id-Related1、Id-Related2
        edges.append((node_id, related_node_1))
        edges.append((node_id, related_node_2))

# 向图中添加所有顶点
g.add_vertices(list(nodes))
# 向图中添加所有边（igraph会自动匹配顶点名称）
g.add_edges(edges)

# 可选：给顶点添加Category属性（如果需要后续分析/可视化区分）
# 注意：如果一个节点对应多个Category，需自行决定处理逻辑（比如取第一个/合并）
# vertex_categories = {}
# with open('your_data.txt', 'r') as f:
#     for line in f:
#         line = line.strip()
#         if not line:
#             continue
#         parts = line.split()
#         node_id = parts[0]
#         if node_id not in vertex_categories:
#             vertex_categories[node_id] = ' '.join(parts[1:-2])
# g.vs['category'] = [vertex_categories.get(v['name'], '') for v in g.vs]

3. 绘制图形 #

用igraph自带的plot()函数就能生成可视化图，你可以根据需求调整布局、大小等参数：

# 选择布局算法，kamada-kawai适合中小型图的美观布局
layout = g.layout('kk')

# 绘制并显示图形
plot(
    g,
    layout=layout,
    vertex_label=g.vs['name'],  # 显示节点名称
    vertex_size=20,
    edge_width=1,
    margin=50  # 避免边缘节点被截断
)

一些实用提示 #

• 若你的图是有向图，初始化Graph时把directed=False改成directed=True即可
• 如果节点数量过多，顶点标签可能重叠，你可以调小vertex_label_size，或者直接关闭标签（vertex_label=None）
• 若要按Category区分节点颜色，可以给顶点添加color属性，比如根据不同Category映射不同RGB值
• 确保TXT文件路径正确，或者把文件放在脚本同一目录下

内容的提问来源于stack exchange，提问作者Lucy