在Python中打印图的连通组件

Question

我试图打印图的连通组件。但它打印出了一个生成器对象。

这是我的graph.py：

import networkx as nx
import matplotlib.pyplot as plt
#import math
import csv
#import random as rand
import sys

def buildG(G, file_, delimiter_):
    #construct the weighted version of the contact graph from cgraph.dat file
    reader = csv.reader(open(file_), delimiter=delimiter_)
    for line in reader:
        if float(line[2]) != 0.0:
            G.add_edge(int(line[0]),int(line[1]),weight=float(line[2]))

def main():
    graph_fn="tempset3.txt";
    G = nx.Graph()  #let's create the graph first
    buildG(G, graph_fn, ',')

    print G.nodes()
    print G.number_of_nodes()

    #nx.draw(G)
    #plt.show(G)

    n = G.number_of_nodes()
    print ("no of nodes: ", n)
    comps=nx.connected_components(G)
    print comps

main()

这是我的tempset3.txt文件

0,1,9
1,3,5
1,4,17824
2,5,1199
2,6,729
5,7,619
5,8,241
5,10,227
7,8,4

当我运行它时，它会显示：

[0, 1, 2, 3, 4, 5, 6, 7, 8, 10]
10
('no of nodes: ', 10)
<generator object connected_components at 0x360f140> 

如何正确打印连通组件？

输出应为: [[0, 1, 3, 4], [2, 5, 6, 7, 8, 10]]。

- suri

Answer 1

只需在生成器对象上使用 list，即可打印出 print (list(comps))

[0, 1, 2, 3, 4, 5, 6, 7, 8, 10]
10
no of nodes:  10
[[0, 1, 3, 4], [2, 5, 6, 7, 8, 10]]

或者遍历生成器对象：

for comp in comps:
        print (comp)
[0, 1, 3, 4]
[2, 5, 6, 7, 8, 10]

- Padraic Cunningham