get_communities.py

import collections
import sys
import time

import community
import networkx as nx

filename = sys.argv[1]

print("Loading graph", time.process_time())
g = nx.read_adjlist(filename)

print("Partitioning graph into communities", time.process_time())
partition = community.best_partition(g)

print("Processing communities", time.process_time())
print("Modularity", community.modularity(partition, g))
communities = collections.defaultdict(set)
for node in partition:
  comm = partition[node]
  communities[comm].add(node)

print("Num communities:", len(communities))
max_comm = max(len(communities[comm]) for comm in communities)
min_comm = min(len(communities[comm]) for comm in communities)
print("Community sizes:", min_comm, "to", max_comm)