如何基于已上传边文件的数据集在Colab中创建网络图

1. 确认已上传的边文件信息 #

在Colab代码块运行以下命令，查看当前运行环境下的所有文件，确认上传的边文件的准确文件名：

!ls

2. 读取边数据集 #

先导入数据处理依赖：

import pandas as pd

根据你的边文件格式，选择对应代码读取数据（默认边文件包含源节点、目标节点两列）：

• 带表头的CSV格式：
  edge_df = pd.read_csv("替换为你的实际边文件名.csv")
• 无表头的制表符分隔TSV/TXT格式：
  edge_df = pd.read_csv("替换为你的实际边文件名.tsv", sep="\t", header=None, names=["source", "target"])

若边文件存在权重、边类型等额外属性，可在读取时保留对应列，后续构建图时可一并导入。若你的边文件中源/目标节点的列名不是source/target，需将后续代码中对应参数替换为你的实际列名。

3. 导入网络构建依赖库 #

Colab默认已预装基础库，若需使用交互可视化工具pyvis，先运行静默安装命令：

!pip install pyvis -q

再导入所需库：

import networkx as nx
from pyvis.network import Network
import matplotlib.pyplot as plt

4. 构建网络图结构 #

根据你的数据关系类型（无向/有向）选择对应图结构：

# 构建无向图（若为有向关系，替换为 G = nx.DiGraph()）
G = nx.from_pandas_edgelist(
    edge_df,
    source="source",
    target="target",
    # 若边有权重，可添加参数：edge_attr="你的权重列名"
)

# 验证图构建结果
print(f"构建完成：节点总数 {G.number_of_nodes()}，边总数 {G.number_of_edges()}")

若输出的节点、边数与预期数据集规模一致，即代表基础图结构构建成功。

5. 在Colab中展示网络图 #

可按需选择静态/交互两种可视化方式：

方式1：静态可视化（快速查看整体结构） #

plt.figure(figsize=(12, 8))
# 可按需调整节点大小、颜色、字体等参数
nx.draw(G, with_labels=True, node_size=500, node_color="lightblue", font_size=10)
plt.show()

方式2：交互可视化（支持缩放、拖拽节点，适合中大规模网络） #

# 初始化适配Colab的交互网络配置
net = Network(notebook=True, width="100%", height="700px", bgcolor="#222222", font_color="white")
# 导入已构建的网络图
net.from_nx(G)
# 生成并展示
net.show("custom_network.html")

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