Граф Python Networkx: не рисовать старый граф вместе с новым графом

Question

Ниже мой код:

import networkx as nx
for i in range(2):
    G = nx.DiGraph()
    if i==0:
        G.add_edge("A", "B")
    elif i==1:
        G.add_edge("A", "C")
    import matplotlib.pyplot as plt
    nx.draw(G)
    plt.savefig(str(i)+".png")
    G.clear()

Следует нарисовать линию AB в файле 0.png и нарисовать линию AC в файле 1.png. Но после того как я его запустил. В 0.png есть одна строка AB, но в 1.png есть две строки: AB и AC. Кажется, что память для 0.png не очищена, хотя у меня было "G.clear ()".

Кто-нибудь знает, как это исправить?

Answer 1

У меня есть решение.

Добавьте plt.clf() после plt.savefig(str(i) + ".png").Он может очистить старый граф в pyplot.Я надеюсь, что это может кому-то помочь.