Graph Partitioning Python