python - 使用 __repr__ 等返回 matplotlib Fig

Question

我正在构建一个用于构建 matplotlib 图表的 Python 类。我选择使用类是因为我希望能够通过一系列步骤构建图表，而不是使用需要大量参数的单个函数。

我的主要工作环境是 Jupyter Lab，我通常关闭交互式绘图，即 plt.ioff()。相反，我喜欢使用 display() 或通过将其放置在单元格的最后一行来隐式显示该图形。

class Example:

    def __init__(self):

        self.fig, self.ax = plt.subplots()

使用上面的例子，我知道我可以做这样的事情

display(Example().fig)

但是，我希望有一种方法可以使 .fig 隐式，这样我就可以执行 display(Example()) 。我原本以为我可以做到

def __repr__():
    return self.fig

但这种方法似乎行不通。有办法做到这一点吗？

Answer 1

我认为你想显示图形的 png 表示，所以

class Example:

    def __init__(self):
        self.fig, self.ax = plt.subplots()

    def _repr_png_(self):
        return display(self.fig)