Как построить несколько картинок с Pylab в Python?

Question

У меня есть следующий код:

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 другое изображение, но каждое следующее изображение содержит строки предыдущего. как я могу предотвратить это? мне нужны отдельные картинки без линий прежней.

Answer 1

Чтобы сделать то, что вы хотите, вам нужно сделать одну из двух вещей:

1. используйте методы clf() и cla() (соответственно очистить фигуру и очистить оси) после сохранения ваших рисунков.

2. Создайте несколько фигур , а затем установите заголовок и метку оси на каждой фигуре. На данный момент вы используете фоновую фигуру по умолчанию, которую дает вам 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")