Python 将在循环中创建的绘图保存到不同的文件中

我编写了一个脚本来绘制pandas.DataFrame的一些列。它是：

import pandas as pd
import matplotlib.pyplot as plt
df=pd.read_csv('alcohol_cons_18', sep=',') #this is the loaded data
a_list=list(df.columns.values) #a list of its columns
for ielement in range(len(a_list)): #iterate over columns and make charts
    print ("column name is %s" %(a_list[ielement]))
    if a_list[ielement]!= 'age':
        df.plot(x='age', y=a_list[ielement])
        plt.savefig(a_list[ielement] + '.png') #savefigures

但脚本以累加的方式保存数字。所以在第二个图中有第一个和第二个图形，依此类推。。 知道如何修改脚本以将这些图形保存在不同的文件中吗？

根据是否要保存某些图形部分，可以选择

plt.clf（）

或

plt.cla（）

。

是否使用

plt.close()

如文件所述，解决您的问题

否则，您可以尝试将绘图指定给枚举图形，如下所示

for ielement in range(len(a_list)): #iterate over columns and make charts
    print ("column name is %s" %(a_list[ielement]))
    if a_list[ielement]!= 'age':
        fig = plt.figure(ielement)
        df.plot(x='age', y=a_list[ielement])
        plt.savefig(a_list[ielement] + '.png')

---

不确定代码中的差异，但plt.close（）是缺少的部分。谢谢。