Python networkx graph：不要将旧图与新图一起绘制

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()

它应该在文件 0.png 中画线 AB，在文件 1.png 中画线 AC。但是，在我运行它之后。在 0.png 中，有一条 AB 线，但在 1.png 中，有两条线：AB 和 AC。虽然我有“G.clear()”，但似乎没有清理 0.png 的内存。

有人知道怎么解决吗？

Answer 1

我有办法了。

在plt.savefig(str(i) + ".png") 之后添加plt.clf()。它可以清理 pyplot 中的旧图。我希望它可以帮助某人。