如何在python中使用pylab绘制多张图片?

时间:2011-07-11 07:54:21

标签: python matplotlib

我有以下代码:

import psycopg2
import pylab


class Datenanzeige:

def __init__(self):
    self.conn = psycopg2.connect("dbname='test' user='test' host='test' password='test'")
    self.cur = self.conn.cursor()

    self.site = ['FFF', 'PPP', 'DDD', 'KKK']
    self.plant = [1, 2, 3, 4, 5, 6]


def anzeige(self):

    for i in range(4):
        for j in range(6):

            self.datProd1 = []
            self.datProd2 = []

            self.cur.execute(""" 
                    SELECT proddate, Sum(mods) AS Summevonmods 
                    FROM "20091229global" 
                    GROUP BY proddate, site, plant, pid 
                    HAVING proddate > Date('today') - 14
                    AND
                    site='"""+self.site[i]+"""'
                    AND
                    pid=802
                    AND
                    plant='"""+str(self.plant[j])+"""' 
                    ORDER BY proddate, site, plant, pid; 
                    """)

            self.row = self.cur.fetchone()

            while self.row:
                self.datProd1.append(self.row[0])
                self.datProd2.append(self.row[1])

                pylab.title('Progress of %s' %self.site[i])
                pylab.ylabel('number')
                pylab.xlabel('date')
                pylab.plot(self.datProd1, self.datProd2)
                pylab.savefig("bild%i" %i )

                self.row=self.cur.fetchone()

q = Datenanzeige()
q.anzeige()

代码正常运行。我得到每个self.site变量另一张图片,但每个后续图片包括前一个图片的行。我怎么能阻止这个? 我需要没有前一行的单独图片。

1 个答案:

答案 0 :(得分:1)

要做你想做的事,你需要做以下两件事之一:

  1. 保存数据后,请使用clf()cla()(清晰的数字和清除轴)方法。

  2. 创建多个figures,然后在每个图上设置标题和轴标签。目前,您正在使用pylab为您提供的默认背景图。您需要明确地创建和修改不同的。我不能写出我头顶的确切代码来做到这一点,但是看看matplotlib api应该可以找到你想要的东西。它看起来像这样:

    import matplotlib.pyplot as plt
    
    fig = plt.figure()
    axes = fig.add_subplot(1,1,1)
    axes.set_xlabel("foo")
    axes.set_ylabel("bar")
    axes.set_title("title")
    # ...
    fig.savefig("filename")