matplotlib：创建新图形并保存到文件

Question

我正在尝试使用matplotlib 生成不同的数字：

import matplotlib.pyplot as plt

for item in item_list:
    plt.imshow(item)
    plt.grid(True)
    # generate id
    plt.savefig(filename + id)
    plt.close()

该循环确实生成了许多文件，但它们似乎显示了不同图形的叠加，而如果我将这些项目逐一绘制，它们看起来就大不相同了。

如何确保每个项目都独立绘制并保存到文件中？

Answer 1

您需要创建一个新的图形对象，或者清除轴。

清轴示例代码：

import matplotlib.pyplot as plt
y_data = [[1,1],[2,2],[3,3]]  #just some dummy data
x = [0,1]  
fig,ax = plt.subplots()
for y in y_data:   
    #generate id   
    ax.cla()  #clear the axis
    ax.plot([0,1],y)
    fig.savefig(filename + id)

新图形对象示例：

import matplotlib.pyplot as plt
y_data = [[1,1],[2,2],[3,3]]  #just some dummy data
x = [0,1]
for y in y_data:  
    #generate id  
    fig,ax = plt.subplots()  #create a new figure
    ax.plot(x,y)
    fig.savefig(filename + id)

希望这有助于您入门。