Добавление угловых компонентов поверх холста

Как я уже сказал в комментарии, прочитайте . Как я могу прикрепить функцию pyplot к экземпляру фигуры? для объяснения разницы между интерфейсами OO и state-machine с matplotlib.

Вы должны изменить свои функции построения, чтобы быть чем-то вроде

def plot(..., ax=None, **kwargs):
    if ax is None:
        ax = gca()
    ax.plot(..., **kwargs)