knowledgegraph.py 文件源码

def draw_tier3():
    print 'loading tier3 data'
    loadEntity(tier3filename)
    print 'entity size: ', len(entityList)

    G = nx.Graph()
    for entity in entityList:
        name = entity.name[0].decode('utf8')
        print name
        G.add_node(name)

        for parent in entity.parent:
            pr = parent.decode('utf8')
            G.add_node(pr)
            G.add_edge(pr, name)

        for child in entity.child:
            cn = child.decode('utf8')
            G.add_node(cn)
            G.add_edge(name, cn)

    pos=nx.spring_layout(G) # positions for all nodes
    nx.draw_networkx_edges(G,pos,width=1.0,alpha=0.5)
    # labels
    nx.draw_networkx_labels(G,pos,font_size=10,font_family='sans-serif')
    plt.axis('off')
    plt.show()