Source code for mpl_toolkits.axes_grid1.mpl_axes

import matplotlib.axes as maxes
from matplotlib.artist import Artist
from matplotlib.axis import XAxis, YAxis

[docs]class SimpleChainedObjects:
[docs] def __init__(self, objects): self._objects = objects
[docs] def __getattr__(self, k): _a = SimpleChainedObjects([getattr(a, k) for a in self._objects]) return _a
[docs] def __call__(self, *args, **kwargs): for m in self._objects: m(*args, **kwargs)
[docs]class Axes(maxes.Axes):
[docs] class AxisDict(dict):
[docs] def __init__(self, axes): self.axes = axes super().__init__()
[docs] def __getitem__(self, k): if isinstance(k, tuple): r = SimpleChainedObjects( # super() within a list comprehension needs explicit args. [super(Axes.AxisDict, self).__getitem__(k1) for k1 in k]) return r elif isinstance(k, slice): if k.start is None and k.stop is None and k.step is None: return SimpleChainedObjects(list(self.values())) else: raise ValueError("Unsupported slice") else: return dict.__getitem__(self, k)
[docs] def __call__(self, *v, **kwargs): return maxes.Axes.axis(self.axes, *v, **kwargs)
def _init_axis_artists(self, axes=None): if axes is None: axes = self self._axislines = self.AxisDict(self) self._axislines.update( bottom=SimpleAxisArtist(self.xaxis, 1, self.spines["bottom"]), top=SimpleAxisArtist(self.xaxis, 2, self.spines["top"]), left=SimpleAxisArtist(self.yaxis, 1, self.spines["left"]), right=SimpleAxisArtist(self.yaxis, 2, self.spines["right"])) @property def axis(self): return self._axislines
[docs] def cla(self): super().cla() self._init_axis_artists()
[docs]class SimpleAxisArtist(Artist):
[docs] def __init__(self, axis, axisnum, spine): self._axis = axis self._axisnum = axisnum self.line = spine if isinstance(axis, XAxis): self._axis_direction = ["bottom", "top"][axisnum-1] elif isinstance(axis, YAxis): self._axis_direction = ["left", "right"][axisnum-1] else: raise ValueError( f"axis must be instance of XAxis or YAxis, but got {axis}") super().__init__()
@property def major_ticks(self): tickline = "tick%dline" % self._axisnum return SimpleChainedObjects([getattr(tick, tickline) for tick in self._axis.get_major_ticks()]) @property def major_ticklabels(self): label = "label%d" % self._axisnum return SimpleChainedObjects([getattr(tick, label) for tick in self._axis.get_major_ticks()]) @property def label(self): return self._axis.label
[docs] def set_visible(self, b): self.toggle(all=b) self.line.set_visible(b) self._axis.set_visible(True) super().set_visible(b)
[docs] def set_label(self, txt): self._axis.set_label_text(txt)
[docs] def toggle(self, all=None, ticks=None, ticklabels=None, label=None): if all: _ticks, _ticklabels, _label = True, True, True elif all is not None: _ticks, _ticklabels, _label = False, False, False else: _ticks, _ticklabels, _label = None, None, None if ticks is not None: _ticks = ticks if ticklabels is not None: _ticklabels = ticklabels if label is not None: _label = label tickOn = "tick%dOn" % self._axisnum labelOn = "label%dOn" % self._axisnum if _ticks is not None: tickparam = {tickOn: _ticks} self._axis.set_tick_params(**tickparam) if _ticklabels is not None: tickparam = {labelOn: _ticklabels} self._axis.set_tick_params(**tickparam) if _label is not None: pos = self._axis.get_label_position() if (pos == self._axis_direction) and not _label: self._axis.label.set_visible(False) elif _label: self._axis.label.set_visible(True) self._axis.set_label_position(self._axis_direction)