network.py 文件源码

def save_graphml(G, filename):
    """Saves a network created get_network to graphml file"""

    if "paper" in G.edges()[0]:
        H = networkx.graph.Graph()
        for e in G.edges():
            H.add_edge(e[0], e[1])
            H[e[0]][e[1]]['weight'] = len(G[e[0]][e[1]]['papers'])
        G = H
    networkx.write_graphml(G, filename)