如何让包装器围绕绘图功能创建和保存图形？

Question

我阅读了有关包装器的信息，并想将其用于绘图。我想要一个包装器函数来在我的绘图函数中创建和保存图形。但是我收到如下所示的错误。这是我的代码：

import numpy as np
import matplotlib.pyplot as plt


# figure creating and saving wrapper
def figure_wrapper(func):
    def wrapper(*args,**kwargs):
        fig = plt.figure()

        fig,name = func(fig,*args,**kwargs)

        fig.savefig(name, bbox_inches='tight')
        plt.close(fig)

    return wrapper

# plotting class
class plotting():

    @figure_wrapper
    def plot1(self,fig,x,y):
        name = 'plot1'
        ax = fig.add_subplot()
        ax.plot(x,y)
        return fig, name

    @figure_wrapper
    def scatter1(self,fig,x,y,):
        name = 'scatter1'
        ax = fig.add_subplot()
        ax.scatter(x, y)
        return fig, name

# data for plotting
x = np.linspace(0, 10, 10)
y = np.linspace(20, 30, 10)
x1 = np.linspace(20, 10, 10)
y1 = np.linspace(60, 30, 10)

# execution of class
plotting = plotting()
plotting.plot1(x = x,y= y)
plotting.scatter1(x = x1,y= y1)

我得到的错误：

Traceback (most recent call last):
  File "C:\Users\jerzy\Documents\Test_drawing_saving.py", line 52, in <module>
    plotting.plot1(x = x,y= y)
  File "C:\Users\jerzy\Documents\Test_drawing_saving.py", line 10, in wrapper
    fig,name = func(fig,*args,**kwargs)
  File "C:\Users\jerzy\Documents\Test_drawing_saving.py", line 23, in plot
    ax = fig.add_subplot()
AttributeError: 'plotting' object has no attribute 'add_subplot'

Answer 1

我只是将你的 fig 声明为你的 plotting class 的属性，然后将 x 和 y 传递给你的包装器。请参阅下面的代码：

import numpy as np
import matplotlib.pyplot as plt


# figure creating and saving wrapper
def figure_wrapper(func):
    def wrapper(*args,**kwargs):
        fig,name = func(*args,**kwargs)
        fig.savefig(name, bbox_inches='tight')
        plt.clf()
        plt.close(fig)
    return wrapper

# plotting class
class plotting():
    fig=plt.figure()
    @figure_wrapper
    def plot(self,x,y):
        name = 'plot1'
        ax = self.fig.add_subplot(label=name)
        ax.plot(x,y)
        return self.fig, name

    @figure_wrapper
    def scatter(self,x,y,):
        name = 'scatter1'
        ax = self.fig.add_subplot(label=name)
        ax.scatter(x, y)
        return self.fig, name

# data for plotting
x = np.linspace(0, 10, 10)
y = np.linspace(20, 30, 10)
x1 = np.linspace(20, 10, 10)
y1 = np.linspace(60, 30, 10)

# execution of class
plotting = plotting()
plotting.plot(x,y)
plotting.scatter(x,y)

输出保存了以下两个数字：

如何让包装器围绕绘图功能创建和保存图形？

How to get wrapper creating and saving figure, around plot function?

python

matplotlib

wrapper