社区发现算法——Louvain 算法代码
时间: 2023-08-02 07:08:26 浏览: 244
社区发现算法Louvain详解.docx
5星 · 资源好评率100%
Louvain 算法是一种常用的社区发现算法,它的实现比较简单,下面是 Python 实现代码:
```python
import networkx as nx
import community
# 构建图
G = nx.karate_club_graph()
# 运行 Louvain 算法
partition = community.best_partition(G)
# 输出每个节点所属的社区
for node, community_id in partition.items():
print(f"Node {node} belongs to community {community_id}")
```
其中,`nx.karate_club_graph()` 用于生成一个 Zachary's Karate Club 的图,可以根据需要替换为自己的图。`community.best_partition(G)` 返回一个字典,其中键为节点编号,值为节点所属的社区编号。
阅读全文