如何在pyvis中绘制具有社区结构的图形

Question

我有一个边缘列表，其中的节点是字符串值，比如城市名称。在应用社区检测算法后，我得到了带有标签的城市，即具有相同标签的城市在同一个社区中。我想要可视化城市的图形，在那里，有着相同标签的城市被用相同的颜色着色。有人能帮我吗？我甚至不知道pyvis有这个特性吗？

我能够使用以下代码绘制一个图形：

from pyvis.network import Network
net = Network(height='800px', width='800px', directed=False, notebook=True)
net.from_nx(G)
net.show("example.html")

但我不知道如何根据检测到的社区向节点添加标签。

Answer 1

在创建pyvis图时，您可以通过给社区中的每个节点相同的颜色来给每个社区一个独特的颜色。类似于：

import random
r = lambda: random.randint(0,255)

# Loop over each community
for community in graph:
    communityColorString = '#%02X%02X%02X' % (r(),r(),r())
    
    # Loop over each node 
    for node in community:
        newGraph.add_node(word_a, word_a, title=word_a, color=communityColorString)