Python 从函数返回NetworkX图形

我试图创建一个函数，从一系列边创建一个networkx图，但它实际上并没有返回该图

def write_graph(nodes):
    G = nx.Graph()
    G.add_nodes_from(range(6))
    for move_number in range(len(nodes)):
        if move_number % 2 == 0:
            p = 1
        else:
            p = 2
        move_list = list(node)
        move = move_dict[move_list[move_number]]
        G.add_edge(move[0], move[1], player = p)
    return G

运行此函数后，图G没有任何边或节点

编辑：我的问题是，我调用函数时没有返回一些内容，例如

write_graph('0')

相对于

G = write_graph('0')

---

我想这个例子会帮助你：你想要图形的可视化表示吗？不，我不需要可视化表示，只需要图形作为networkx图形