pyvis

https://github.com/WestHealth/pyvis

pyvis 是一个用于创建和可视化交互式网络图的 Python 库。

pyvis 可以在每个节点或每个边缘的基础上进行定制。可以为节点指定颜色、大小、标签和其他元数据。

每个图形都可以进行交互，允许拖动、悬停以及选择节点和边。

每个图形的布局算法也可以进行调整，以允许尝试渲染更大的图形。

​       

初体验

库的安装

我们可以直接使用 pip 来进行安装。

pip install pyvis

创建网络

首先，使用 pyvis 中的 Network 类创建一个 net。

net = Network(
    notebook=True,
)

要在 Jupyter Notebook 上显示图表，请将 notebook 参数设置为 True。

添加节点

你现在可以向图表添加节点。

net.add_node("Singapore")
net.add_node("San Francisco")
net.add_node("Tokyo")
net.add_nodes(["Riga", "Copenhagen"],
              color=['lightgreen', 'yellow'])

函数 add_node() 用来添加单个节点，而函数 add_nodes() 则添加多个节点到图中。

你还可以为这两个函数设置可选的颜色参数来设置节点的颜色。

要显示图形，请使用 show() 函数。

net.show('mygraph.html')

添加边

将节点添加到图中后，你现在可以添加边来连接节点。

net.add_edge("Singapore","San Francisco") 
net.add_edge("San Francisco","Tokyo")
net.add_edges(
    [
        ("Riga","Copenhagen"),
        ("Copenhagen","Singapore"),
        ("Singapore","Tokyo"),
        ("Riga","San Francisco"),
        ("San Francisco","Singapore"),
    ]
)

net.show('mygraph.html')

函数 add_edge() 添加连接两个节点的单个边，而函数 add_edges() 接受连接各个节点的元组列表。

该图现在显示连接各个节点的边。

尝试拖动每个节点，看看它们在释放后如何被拉回。

有向图

如果你想要一个有向图，你应该在 Network 类中设置有向参数。

net = Network(
    notebook=True,
    directed=True
)

你现在应该看到一个有向图。

修改图形的物理特性

如果单击并拖动图表中的节点，你会注意到节点会四处弹跳。

当你释放鼠标时，节点将弹回其原始位置。

​       

你可以使用 repulsion() 函数自定义图形背后的物理原理（它们如何弹回、弹簧的阻尼等）。

以下语句显示了 repulsion() 函数中所有参数的默认值。

net.repulsion(
    node_distance=100,
    central_gravity=0.2,
    spring_length=200,
    spring_strength=0.05,
    damping=0.09,
)

以下是各个参数的用途。

• node_distance，这是斥力的影响范围。

• central_gravity，将整个网络拉向中心的重力吸引子。

• spring_length，边的剩余长度。

• spring_strength，边缘弹簧的强度。

• 阻尼，范围从 0 到 1 的值，表示上一次物理模拟迭代中的速度有多少会延续到下一次迭代。

pyvis 库最有趣的方面是它使你的网络图变得栩栩如生。交互式网络图非常适合社交网络、公司结构或你想要可视化实体之间关系的其他网络。

来源——程序员小寒