我的Matplotlib子图标题已删除

时间:2019-04-11 08:56:01

标签: python matplotlib

运行代码时,我创建一个图形,然后在该图形中创建一个子图。然后,当我尝试使用ax.set_title("title")向其添加标题时,有时会显示一秒钟,然后消失。我也尝试过使用plot.title并没有运气。

在一个小示例中,我尝试重新创建该错误,但是由于某种原因,该错误在那儿工作得很好,所以这里是代码的整个源代码。

import matplotlib.pyplot as plt
import matplotlib.animation as animation
import matplotlib.style as style
import plotgen
from matplotlib.widgets import Button

class plotWindow():
    def __init__(self):
        style.use("bmh")
        self.dp = 30

        self.fig = plt.figure()
        self.ax = self.fig.add_subplot(1, 1, 1, label="ax1")


        self.cax = 1
        self.maxax = 2
        self.minax = 1

        plotgen.clear("plot1.txt")
        plotgen.clear("plot2.txt")

        axnext = plt.axes([0.80, 0.01, 0.06, 0.06])
        axprev = plt.axes([0.73, 0.01, 0.06, 0.06])

        bnext = Button(axnext, 'Next >')
        bnext.on_clicked(self.changePlotNext)
        bprev = Button(axprev, "< Previous")
        bprev.on_clicked(self.changePlotPrev)


        ani = animation.FuncAnimation(self.fig, self.animate, interval=500)
        plt.show()

    def changePlotNext(self, i):
        if self.cax < self.maxax:
            self.cax += 1
            self.ax.set_title("Pump " + str(self.cax))

    def changePlotPrev(self, i):
        if self.cax > self.minax:
            self.cax -= 1
            self.ax.set_title("Pump " + str(self.cax))

    def animate(self, i):
        if self.cax == 1:
            plotgen.generate("plot1.txt")
            graph_data = open('plot1.txt', 'r').read()
            lines = graph_data.split('\n')
            xs = []
            ys = []
            for line in lines:
                if len(line) > 1:
                    x, y = line.split(',')
                    xs.append(x)
                    ys.append(float(y))
            self.ax.clear()
            lx = len(xs)
            ly = len(ys)
            if len(xs) < self.dp:
                pxs = xs
                pys = ys
            else:
                pxs = xs[(lx - (self.dp - 1)):(lx - 1)]
                pys = ys[(ly - (self.dp - 1)):(ly - 1)]
            self.ax.plot(pxs, pys, "r")
        elif self.cax == 2:
            plotgen.generate("plot2.txt")
            graph_data = open('plot2.txt', 'r').read()
            lines = graph_data.split('\n')
            xs = []
            ys = []
            for line in lines:
                if len(line) > 1:
                    x, y = line.split(',')
                    xs.append(x)
                    ys.append(float(y))
            self.ax.clear()
            lx = len(xs)
            ly = len(ys)
            if len(xs) <= self.dp:
                pxs = xs
                pys = ys
            else:
                pxs = xs[(lx - (self.dp - 1)):(lx - 1)]
                pys = ys[(ly - (self.dp - 1)):(ly - 1)]
            self.ax.plot(pxs, pys)
plotWindow()

正如您在我的changePlotNextchangePlotPrev函数中所看到的那样,我正在尝试更改标题。有时,当我更改时它们会显示一瞬间,但随后消失了。而且我很清楚,在更改情节之前,我还没有设置要显示的标题。

1 个答案:

答案 0 :(得分:1)

animate中,您拥有self.ax.clear(),这将删除轴上的所有艺术家,文本等,包括标题。

然后,一个简单的选择是在清除轴后重置标题。因此,如果您添加:

 self.ax.set_title("Pump " + str(self.cax))

在两个地方,一旦您致电self.ax.clear(),标题仍然会显示。

另一种选择是停止清除轴,而只删除需要删除的项目。我认为这只是您绘制的线条?因此,例如,您可以删除对self.ax.clear()的呼叫,然后添加:

for line in self.ax.lines:
    line.remove()

代替它。这样将只删除绘制的线条,但保留标题。