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

0 投票
1 回答
1148 浏览
提问于 2025-04-16 21:15

我有以下代码:

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 个回答

1

要实现你想要的效果,你需要做以下两件事中的一件:

  1. 在保存好你的图形之后,使用 clf()cla() 方法(分别是清空图形和清空坐标轴)。

  2. 创建多个 图形,然后在每个图形上设置标题和坐标轴标签。目前,你使用的是 pylab 提供的默认背景图形。你需要明确地创建和修改不同的图形。我现在不能直接写出确切的代码,但查看一下 matplotlib 的文档,应该能找到你需要的信息。它的代码大概会是这样的:

    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")
    

撰写回答