Python 的 `pyvis` 是一个用于创建交互式网络图可视化的第三方库，它基于 `NetworkX` 和 `Vis.js` 构建，可以方便地生成 HTML 文件以在浏览器中查看图形结构。`pyvis` 特别适合用于展示社交网络、知识图谱、关系图谱等场景下的节点与边的连接关系。 ### 安装 可以通过 pip 安装 `pyvis`： ```bash pip install pyvis ``` 同时需要安装 `NetworkX` 来构建和操作图结构： ```bash pip install networkx ``` ### 基本用法 以下是一个使用 `pyvis` 创建简单网络图的示例代码： ```python from pyvis.network import Network import networkx as nx # 创建一个 NetworkX 图对象 G = nx.Graph() # 添加节点和边 G.add_node(1, label="Node 1") G.add_node(2, label="Node 2") G.add_edge(1, 2, title="Edge between 1 and 2") # 创建 PyVis 网络对象 net = Network(notebook=True, height="500px", width="100%") # 将 NetworkX 图转换为 PyVis 图 net.from_nx(G) # 保存为 HTML 文件并在浏览器中打开 net.show("example.html") ``` 上述代码创建了一个包含两个节点和一条边的图，并将其保存为名为 `example.html` 的文件。打开该文件可以在浏览器中看到交互式的图形界面，支持缩放、拖动和悬停显示边信息等功能。 ### 高级功能 - **自定义节点样式**：可以设置节点的颜色、形状、大小等属性。 - **添加物理模拟布局**：通过启用 `repulsion` 等选项控制节点之间的排斥力，使图形更美观。 - **导出为 JSON**：使用 `net.save_graph("graph.json")` 可将图结构保存为 JSON 格式，便于后续处理或嵌入网页。 - **集成 Jupyter Notebook**：设置 `notebook=True` 可直接在 Jupyter Notebook 中渲染图形。 ### 示例：带标签和颜色的图 ```python # 自定义节点和边样式 net.add_node(1, label="Alice", color="blue", shape="dot", size=20) net.add_node(2, label="Bob", color="red", shape="ellipse", size=25) net.add_edge(1, 2, title="Knows", color="green") # 启用物理模拟布局 net.set_options(""" var options = { "physics": { "enabled": true, "repulsion": { "distance": 200 } } } """) net.show("customized_example.html") ``` 这段代码展示了如何更改节点的颜色、形状和大小，并设置了边的颜色和标题，还启用了物理模拟来优化图的布局效果。 ### 应用场景 - 社交网络分析 - 知识图谱可视化 - 数据流/依赖关系图 - 项目管理中的任务依赖关系图 - 网络拓扑结构展示 ### 注意事项 - `pyvis` 生成的是 HTML 文件，因此不适合用于纯文本环境下的可视化。 - 如果图形非常复杂，加载速度可能会受到影响，建议对大规模图进行适当简化或分层展示。 ---