Eu tenho uma classe de figura personalizada e gostaria de garantir que todos os eixos associados a ela, sejam criados comsubplots() outwinx(), etc. têm comportamentos personalizados.

No momento, eu faço isso vinculando novos métodos a cada eixo após a criação, por exemplo usando

import types

def my_ax_method(ax, test):
    print('{0} is doing something new as a {1}.'.format(ax, test))

class MyFigure(matplotlib.figure.Figure):
    def __init__(self, **kwargs):
        super(MyFigure, self).__init__(**kwargs)            
        axes_a = None
        axes_b = None
        axes_c = None

    def setup_axes(self, ax):    
        self.axes_a =  ax
        self.axes_b = self.axes_a.twinx()
        self.axes_c = self.axes_a.twiny()    
        self.axes_a.my_method = types.MethodType(my_ax_method, self.axes_a)
        self.axes_b.my_method = types.MethodType(my_ax_method, self.axes_b)
        self.axes_c.my_method = types.MethodType(my_ax_method, self.axes_c)

em algo como

    fig, ax = matplotlib.pyplot.subplots(FigureClass=MyFigure)
    fig.setup_axes(ax)

    fig.axes_a.my_method("probe of A")
    fig.axes_b.my_method("test of B")
    fig.axes_c.my_method("trial of C")

Parece uma maneira frágil de realizar o que estou tentando fazer. Existe uma maneira melhor e mais pitônica de fazer isso?

Em particular, existe uma maneira de garantir que todos osAxes do meu costumeFigure classe são de uma classe específica (como é feito para as próprias figuras): um costumeAxes classe própria que poderia ter esses métodos como parte de sua definição?