在制作新情节时再次绘制旧数据(对于早期的情节)

时间:2013-06-08 12:39:49

标签: python matplotlib

我的CSV文件包含以下条目

id, A, B    #first line.
1, 1 2 3 4 5 6 7 8(and many more), 0 9 8 7 6 5 4 3 2(and many more) #line 1
...
...
(many more lines like this)

以下是每行绘制图形并将其保存到文件的脚本。例如,要绘制的对将是{1,0}; {2,9}; {3,8}; ...

import numpy as np
import matplotlib.pyplot as plt
import csv

def plot( fileName, x, y):
    fileNameWithExt=fileName
    fileNameWithExt+='.png'
    print fileNameWithExt
    plt.plot( x, y )
    plt.xlabel( "X values" )
    plt.ylabel( "Y values" )
    plt.savefig(fileNameWithExt)

if __name__=="__main__":
    with open('test', 'r')  as csvfile:
        next(csvfile)
        spamreader = csv.reader(csvfile, delimiter=',')
        for row in spamreader:
            print (row[1].strip()).split(" ")
            x = map(int, (row[1].strip()).split(" "))
            y = map(int, (row[2].strip()).split(" "))
            plot(row[0], x, y)

示例文件(test)如下:

id, A, B
train1, 1 2 3 4 5 6 7 8, 2 9 8 7 6 5 4 3
train2, 8 5 6 9 3 2 0 1, 2 4 6 8 6 9 1 6

train1.png train2.png

正如人们所看到的,第二个图也包含第一个图的值。即使我在绘图后删除了列表,我也不明白为什么会发生这种情况。我犯了什么错误。

任何帮助表示感谢。

1 个答案:

答案 0 :(得分:2)

您需要关闭或清除循环之间的图:

例如。 plt.close()

cla()清除轴,即当前图中当前活动的轴。它使其他轴保持不变。

clf()用它的所有轴清除整个当前数字,但是打开窗口,这样它就可以重复用于其他图。

close()关闭一个窗口,如果没有另外指定,它将是当前窗口。