我有一个自定义图形类,并希望确保与其关联的所有轴,无论是使用subplots()
or twinx()
等具有自定义行为。
现在,我通过在创建每个轴后将新方法绑定到每个轴来实现这一点,例如通过使用
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)
在类似的事情中
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")
这似乎是完成我想做的事情的一种脆弱的方式。有没有更好、更 Pythonic 的方法来解决这个问题?
特别是,有没有一种方法可以确保所有Axes
我的习惯Figure
类属于特定类(就像对数字本身所做的那样 https://stackoverflow.com/a/27470938/656912): 一个习俗Axes
我自己的类可以将这些方法作为其定义的一部分吗?