""" Classes for the ticks and x- and y-axis. """ import datetime import functools import logging from numbers import Real import warnings import numpy as np import matplotlib as mpl from matplotlib import _api, cbook import matplotlib.artist as martist import matplotlib.colors as mcolors import matplotlib.lines as mlines import matplotlib.scale as mscale import matplotlib.text as mtext import matplotlib.ticker as mticker import matplotlib.transforms as mtransforms import matplotlib.units as munits _log = logging.getLogger(__name__) GRIDLINE_INTERPOLATION_STEPS = 180 # This list is being used for compatibility with Axes.grid, which # allows all Line2D kwargs. _line_inspector = martist.ArtistInspector(mlines.Line2D) _line_param_names = _line_inspector.get_setters() _line_param_aliases = [list(d)[0] for d in _line_inspector.aliasd.values()] _gridline_param_names = ['grid_' + name for name in _line_param_names + _line_param_aliases] class Tick(martist.Artist): """ Abstract base class for the axis ticks, grid lines and labels. Ticks mark a position on an Axis. They contain two lines as markers and two labels; one each for the bottom and top positions (in case of an `.XAxis`) or for the left and right positions (in case of a `.YAxis`). Attributes ---------- tick1line : `~matplotlib.lines.Line2D` The left/bottom tick marker. tick2line : `~matplotlib.lines.Line2D` The right/top tick marker. gridline : `~matplotlib.lines.Line2D` The grid line associated with the label position. label1 : `~matplotlib.text.Text` The left/bottom tick label. label2 : `~matplotlib.text.Text` The right/top tick label. """ def __init__( self, axes, loc, *, size=None, # points width=None, color=None, tickdir=None, pad=None, labelsize=None, labelcolor=None, labelfontfamily=None, zorder=None, gridOn=None, # defaults to axes.grid depending on axes.grid.which tick1On=True, tick2On=True, label1On=True, label2On=False, major=True, labelrotation=0, grid_color=None, grid_linestyle=None, grid_linewidth=None, grid_alpha=None, **kwargs, # Other Line2D kwargs applied to gridlines. ): """ bbox is the Bound2D bounding box in display coords of the Axes loc is the tick location in data coords size is the tick size in points """ super().__init__() if gridOn is None: which = mpl.rcParams['axes.grid.which'] if major and (which in ('both', 'major')): gridOn = mpl.rcParams['axes.grid'] elif (not major) and (which in ('both', 'minor')): gridOn = mpl.rcParams['axes.grid'] else: gridOn = False self.set_figure(axes.figure) self.axes = axes self._loc = loc self._major = major name = self.__name__ major_minor = "major" if major else "minor" if size is None: size = mpl.rcParams[f"{name}.{major_minor}.size"] self._size = size if width is None: width = mpl.rcParams[f"{name}.{major_minor}.width"] self._width = width if color is None: color = mpl.rcParams[f"{name}.color"] if pad is None: pad = mpl.rcParams[f"{name}.{major_minor}.pad"] self._base_pad = pad if labelcolor is None: labelcolor = mpl.rcParams[f"{name}.labelcolor"] if cbook._str_equal(labelcolor, 'inherit'): # inherit from tick color labelcolor = mpl.rcParams[f"{name}.color"] if labelsize is None: labelsize = mpl.rcParams[f"{name}.labelsize"] self._set_labelrotation(labelrotation) if zorder is None: if major: zorder = mlines.Line2D.zorder + 0.01 else: zorder = mlines.Line2D.zorder self._zorder = zorder grid_color = mpl._val_or_rc(grid_color, "grid.color") grid_linestyle = mpl._val_or_rc(grid_linestyle, "grid.linestyle") grid_linewidth = mpl._val_or_rc(grid_linewidth, "grid.linewidth") if grid_alpha is None and not mcolors._has_alpha_channel(grid_color): # alpha precedence: kwarg > color alpha > rcParams['grid.alpha'] # Note: only resolve to rcParams if the color does not have alpha # otherwise `grid(color=(1, 1, 1, 0.5))` would work like # grid(color=(1, 1, 1, 0.5), alpha=rcParams['grid.alpha']) # so the that the rcParams default would override color alpha. grid_alpha = mpl.rcParams["grid.alpha"] grid_kw = {k[5:]: v for k, v in kwargs.items()} self.tick1line = mlines.Line2D( [], [], color=color, linestyle="none", zorder=zorder, visible=tick1On, markeredgecolor=color, markersize=size, markeredgewidth=width, ) self.tick2line = mlines.Line2D( [], [], color=color, linestyle="none", zorder=zorder, visible=tick2On, markeredgecolor=color, markersize=size, markeredgewidth=width, ) self.gridline = mlines.Line2D( [], [], color=grid_color, alpha=grid_alpha, visible=gridOn, linestyle=grid_linestyle, linewidth=grid_linewidth, marker="", **grid_kw, ) self.gridline.get_path()._interpolation_steps = \ GRIDLINE_INTERPOLATION_STEPS self.label1 = mtext.Text( np.nan, np.nan, fontsize=labelsize, color=labelcolor, visible=label1On, fontfamily=labelfontfamily, rotation=self._labelrotation[1]) self.label2 = mtext.Text( np.nan, np.nan, fontsize=labelsize, color=labelcolor, visible=label2On, fontfamily=labelfontfamily, rotation=self._labelrotation[1]) self._apply_tickdir(tickdir) for artist in [self.tick1line, self.tick2line, self.gridline, self.label1, self.label2]: self._set_artist_props(artist) self.update_position(loc) def _set_labelrotation(self, labelrotation): if isinstance(labelrotation, str): mode = labelrotation angle = 0 elif isinstance(labelrotation, (tuple, list)): mode, angle = labelrotation else: mode = 'default' angle = labelrotation _api.check_in_list(['auto', 'default'], labelrotation=mode) self._labelrotation = (mode, angle) @property def _pad(self): return self._base_pad + self.get_tick_padding() def _apply_tickdir(self, tickdir): """Set tick direction. Valid values are 'out', 'in', 'inout'.""" # This method is responsible for verifying input and, in subclasses, for setting # the tick{1,2}line markers. From the user perspective this should always be # called through _apply_params, which further updates ticklabel positions using # the new pads. if tickdir is None: tickdir = mpl.rcParams[f'{self.__name__}.direction'] else: _api.check_in_list(['in', 'out', 'inout'], tickdir=tickdir) self._tickdir = tickdir def get_tickdir(self): return self._tickdir def get_tick_padding(self): """Get the length of the tick outside of the Axes.""" padding = { 'in': 0.0, 'inout': 0.5, 'out': 1.0 } return self._size * padding[self._tickdir] def get_children(self): children = [self.tick1line, self.tick2line, self.gridline, self.label1, self.label2] return children @_api.rename_parameter("3.8", "clippath", "path") def set_clip_path(self, path, transform=None): # docstring inherited super().set_clip_path(path, transform) self.gridline.set_clip_path(path, transform) self.stale = True def contains(self, mouseevent): """ Test whether the mouse event occurred in the Tick marks. This function always returns false. It is more useful to test if the axis as a whole contains the mouse rather than the set of tick marks. """ return False, {} def set_pad(self, val): """ Set the tick label pad in points Parameters ---------- val : float """ self._apply_params(pad=val) self.stale = True def get_pad(self): """Get the value of the tick label pad in points.""" return self._base_pad def get_loc(self): """Return the tick location (data coords) as a scalar.""" return self._loc @martist.allow_rasterization def draw(self, renderer): if not self.get_visible(): self.stale = False return renderer.open_group(self.__name__, gid=self.get_gid()) for artist in [self.gridline, self.tick1line, self.tick2line, self.label1, self.label2]: artist.draw(renderer) renderer.close_group(self.__name__) self.stale = False @_api.deprecated("3.8") def set_label1(self, s): """ Set the label1 text. Parameters ---------- s : str """ self.label1.set_text(s) self.stale = True set_label = set_label1 @_api.deprecated("3.8") def set_label2(self, s): """ Set the label2 text. Parameters ---------- s : str """ self.label2.set_text(s) self.stale = True def set_url(self, url): """ Set the url of label1 and label2. Parameters ---------- url : str """ super().set_url(url) self.label1.set_url(url) self.label2.set_url(url) self.stale = True def _set_artist_props(self, a): a.set_figure(self.figure) def get_view_interval(self): """ Return the view limits ``(min, max)`` of the axis the tick belongs to. """ raise NotImplementedError('Derived must override') def _apply_params(self, **kwargs): for name, target in [("gridOn", self.gridline), ("tick1On", self.tick1line), ("tick2On", self.tick2line), ("label1On", self.label1), ("label2On", self.label2)]: if name in kwargs: target.set_visible(kwargs.pop(name)) if any(k in kwargs for k in ['size', 'width', 'pad', 'tickdir']): self._size = kwargs.pop('size', self._size) # Width could be handled outside this block, but it is # convenient to leave it here. self._width = kwargs.pop('width', self._width) self._base_pad = kwargs.pop('pad', self._base_pad) # _apply_tickdir uses _size and _base_pad to make _pad, and also # sets the ticklines markers. self._apply_tickdir(kwargs.pop('tickdir', self._tickdir)) for line in (self.tick1line, self.tick2line): line.set_markersize(self._size) line.set_markeredgewidth(self._width) # _get_text1_transform uses _pad from _apply_tickdir. trans = self._get_text1_transform()[0] self.label1.set_transform(trans) trans = self._get_text2_transform()[0] self.label2.set_transform(trans) tick_kw = {k: v for k, v in kwargs.items() if k in ['color', 'zorder']} if 'color' in kwargs: tick_kw['markeredgecolor'] = kwargs['color'] self.tick1line.set(**tick_kw) self.tick2line.set(**tick_kw) for k, v in tick_kw.items(): setattr(self, '_' + k, v) if 'labelrotation' in kwargs: self._set_labelrotation(kwargs.pop('labelrotation')) self.label1.set(rotation=self._labelrotation[1]) self.label2.set(rotation=self._labelrotation[1]) label_kw = {k[5:]: v for k, v in kwargs.items() if k in ['labelsize', 'labelcolor', 'labelfontfamily']} self.label1.set(**label_kw) self.label2.set(**label_kw) grid_kw = {k[5:]: v for k, v in kwargs.items() if k in _gridline_param_names} self.gridline.set(**grid_kw) def update_position(self, loc): """Set the location of tick in data coords with scalar *loc*.""" raise NotImplementedError('Derived must override') def _get_text1_transform(self): raise NotImplementedError('Derived must override') def _get_text2_transform(self): raise NotImplementedError('Derived must override') class XTick(Tick): """ Contains all the Artists needed to make an x tick - the tick line, the label text and the grid line """ __name__ = 'xtick' def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) # x in data coords, y in axes coords ax = self.axes self.tick1line.set( data=([0], [0]), transform=ax.get_xaxis_transform("tick1")) self.tick2line.set( data=([0], [1]), transform=ax.get_xaxis_transform("tick2")) self.gridline.set( data=([0, 0], [0, 1]), transform=ax.get_xaxis_transform("grid")) # the y loc is 3 points below the min of y axis trans, va, ha = self._get_text1_transform() self.label1.set( x=0, y=0, verticalalignment=va, horizontalalignment=ha, transform=trans, ) trans, va, ha = self._get_text2_transform() self.label2.set( x=0, y=1, verticalalignment=va, horizontalalignment=ha, transform=trans, ) def _get_text1_transform(self): return self.axes.get_xaxis_text1_transform(self._pad) def _get_text2_transform(self): return self.axes.get_xaxis_text2_transform(self._pad) def _apply_tickdir(self, tickdir): # docstring inherited super()._apply_tickdir(tickdir) mark1, mark2 = { 'out': (mlines.TICKDOWN, mlines.TICKUP), 'in': (mlines.TICKUP, mlines.TICKDOWN), 'inout': ('|', '|'), }[self._tickdir] self.tick1line.set_marker(mark1) self.tick2line.set_marker(mark2) def update_position(self, loc): """Set the location of tick in data coords with scalar *loc*.""" self.tick1line.set_xdata((loc,)) self.tick2line.set_xdata((loc,)) self.gridline.set_xdata((loc,)) self.label1.set_x(loc) self.label2.set_x(loc) self._loc = loc self.stale = True def get_view_interval(self): # docstring inherited return self.axes.viewLim.intervalx class YTick(Tick): """ Contains all the Artists needed to make a Y tick - the tick line, the label text and the grid line """ __name__ = 'ytick' def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) # x in axes coords, y in data coords ax = self.axes self.tick1line.set( data=([0], [0]), transform=ax.get_yaxis_transform("tick1")) self.tick2line.set( data=([1], [0]), transform=ax.get_yaxis_transform("tick2")) self.gridline.set( data=([0, 1], [0, 0]), transform=ax.get_yaxis_transform("grid")) # the y loc is 3 points below the min of y axis trans, va, ha = self._get_text1_transform() self.label1.set( x=0, y=0, verticalalignment=va, horizontalalignment=ha, transform=trans, ) trans, va, ha = self._get_text2_transform() self.label2.set( x=1, y=0, verticalalignment=va, horizontalalignment=ha, transform=trans, ) def _get_text1_transform(self): return self.axes.get_yaxis_text1_transform(self._pad) def _get_text2_transform(self): return self.axes.get_yaxis_text2_transform(self._pad) def _apply_tickdir(self, tickdir): # docstring inherited super()._apply_tickdir(tickdir) mark1, mark2 = { 'out': (mlines.TICKLEFT, mlines.TICKRIGHT), 'in': (mlines.TICKRIGHT, mlines.TICKLEFT), 'inout': ('_', '_'), }[self._tickdir] self.tick1line.set_marker(mark1) self.tick2line.set_marker(mark2) def update_position(self, loc): """Set the location of tick in data coords with scalar *loc*.""" self.tick1line.set_ydata((loc,)) self.tick2line.set_ydata((loc,)) self.gridline.set_ydata((loc,)) self.label1.set_y(loc) self.label2.set_y(loc) self._loc = loc self.stale = True def get_view_interval(self): # docstring inherited return self.axes.viewLim.intervaly class Ticker: """ A container for the objects defining tick position and format. Attributes ---------- locator : `~matplotlib.ticker.Locator` subclass Determines the positions of the ticks. formatter : `~matplotlib.ticker.Formatter` subclass Determines the format of the tick labels. """ def __init__(self): self._locator = None self._formatter = None self._locator_is_default = True self._formatter_is_default = True @property def locator(self): return self._locator @locator.setter def locator(self, locator): if not isinstance(locator, mticker.Locator): raise TypeError('locator must be a subclass of ' 'matplotlib.ticker.Locator') self._locator = locator @property def formatter(self): return self._formatter @formatter.setter def formatter(self, formatter): if not isinstance(formatter, mticker.Formatter): raise TypeError('formatter must be a subclass of ' 'matplotlib.ticker.Formatter') self._formatter = formatter class _LazyTickList: """ A descriptor for lazy instantiation of tick lists. See comment above definition of the ``majorTicks`` and ``minorTicks`` attributes. """ def __init__(self, major): self._major = major def __get__(self, instance, owner): if instance is None: return self else: # instance._get_tick() can itself try to access the majorTicks # attribute (e.g. in certain projection classes which override # e.g. get_xaxis_text1_transform). In order to avoid infinite # recursion, first set the majorTicks on the instance to an empty # list, then create the tick and append it. if self._major: instance.majorTicks = [] tick = instance._get_tick(major=True) instance.majorTicks.append(tick) return instance.majorTicks else: instance.minorTicks = [] tick = instance._get_tick(major=False) instance.minorTicks.append(tick) return instance.minorTicks class Axis(martist.Artist): """ Base class for `.XAxis` and `.YAxis`. Attributes ---------- isDefault_label : bool axes : `~matplotlib.axes.Axes` The `~.axes.Axes` to which the Axis belongs. major : `~matplotlib.axis.Ticker` Determines the major tick positions and their label format. minor : `~matplotlib.axis.Ticker` Determines the minor tick positions and their label format. callbacks : `~matplotlib.cbook.CallbackRegistry` label : `~matplotlib.text.Text` The axis label. labelpad : float The distance between the axis label and the tick labels. Defaults to :rc:`axes.labelpad` = 4. offsetText : `~matplotlib.text.Text` A `.Text` object containing the data offset of the ticks (if any). pickradius : float The acceptance radius for containment tests. See also `.Axis.contains`. majorTicks : list of `.Tick` The major ticks. .. warning:: Ticks are not guaranteed to be persistent. Various operations can create, delete and modify the Tick instances. There is an imminent risk that changes to individual ticks will not survive if you work on the figure further (including also panning/zooming on a displayed figure). Working on the individual ticks is a method of last resort. Use `.set_tick_params` instead if possible. minorTicks : list of `.Tick` The minor ticks. """ OFFSETTEXTPAD = 3 # The class used in _get_tick() to create tick instances. Must either be # overwritten in subclasses, or subclasses must reimplement _get_tick(). _tick_class = None def __str__(self): return "{}({},{})".format( type(self).__name__, *self.axes.transAxes.transform((0, 0))) def __init__(self, axes, *, pickradius=15, clear=True): """ Parameters ---------- axes : `~matplotlib.axes.Axes` The `~.axes.Axes` to which the created Axis belongs. pickradius : float The acceptance radius for containment tests. See also `.Axis.contains`. clear : bool, default: True Whether to clear the Axis on creation. This is not required, e.g., when creating an Axis as part of an Axes, as ``Axes.clear`` will call ``Axis.clear``. .. versionadded:: 3.8 """ super().__init__() self._remove_overlapping_locs = True self.set_figure(axes.figure) self.isDefault_label = True self.axes = axes self.major = Ticker() self.minor = Ticker() self.callbacks = cbook.CallbackRegistry(signals=["units"]) self._autolabelpos = True self.label = mtext.Text( np.nan, np.nan, fontsize=mpl.rcParams['axes.labelsize'], fontweight=mpl.rcParams['axes.labelweight'], color=mpl.rcParams['axes.labelcolor'], ) self._set_artist_props(self.label) self.offsetText = mtext.Text(np.nan, np.nan) self._set_artist_props(self.offsetText) self.labelpad = mpl.rcParams['axes.labelpad'] self.pickradius = pickradius # Initialize here for testing; later add API self._major_tick_kw = dict() self._minor_tick_kw = dict() if clear: self.clear() else: self.converter = None self.units = None self._autoscale_on = True @property def isDefault_majloc(self): return self.major._locator_is_default @isDefault_majloc.setter def isDefault_majloc(self, value): self.major._locator_is_default = value @property def isDefault_majfmt(self): return self.major._formatter_is_default @isDefault_majfmt.setter def isDefault_majfmt(self, value): self.major._formatter_is_default = value @property def isDefault_minloc(self): return self.minor._locator_is_default @isDefault_minloc.setter def isDefault_minloc(self, value): self.minor._locator_is_default = value @property def isDefault_minfmt(self): return self.minor._formatter_is_default @isDefault_minfmt.setter def isDefault_minfmt(self, value): self.minor._formatter_is_default = value def _get_shared_axes(self): """Return Grouper of shared Axes for current axis.""" return self.axes._shared_axes[ self._get_axis_name()].get_siblings(self.axes) def _get_shared_axis(self): """Return list of shared axis for current axis.""" name = self._get_axis_name() return [ax._axis_map[name] for ax in self._get_shared_axes()] def _get_axis_name(self): """Return the axis name.""" return [name for name, axis in self.axes._axis_map.items() if axis is self][0] # During initialization, Axis objects often create ticks that are later # unused; this turns out to be a very slow step. Instead, use a custom # descriptor to make the tick lists lazy and instantiate them as needed. majorTicks = _LazyTickList(major=True) minorTicks = _LazyTickList(major=False) def get_remove_overlapping_locs(self): return self._remove_overlapping_locs def set_remove_overlapping_locs(self, val): self._remove_overlapping_locs = bool(val) remove_overlapping_locs = property( get_remove_overlapping_locs, set_remove_overlapping_locs, doc=('If minor ticker locations that overlap with major ' 'ticker locations should be trimmed.')) def set_label_coords(self, x, y, transform=None): """ Set the coordinates of the label. By default, the x coordinate of the y label and the y coordinate of the x label are determined by the tick label bounding boxes, but this can lead to poor alignment of multiple labels if there are multiple Axes. You can also specify the coordinate system of the label with the transform. If None, the default coordinate system will be the axes coordinate system: (0, 0) is bottom left, (0.5, 0.5) is center, etc. """ self._autolabelpos = False if transform is None: transform = self.axes.transAxes self.label.set_transform(transform) self.label.set_position((x, y)) self.stale = True def get_transform(self): """Return the transform used in the Axis' scale""" return self._scale.get_transform() def get_scale(self): """Return this Axis' scale (as a str).""" return self._scale.name def _set_scale(self, value, **kwargs): if not isinstance(value, mscale.ScaleBase): self._scale = mscale.scale_factory(value, self, **kwargs) else: self._scale = value self._scale.set_default_locators_and_formatters(self) self.isDefault_majloc = True self.isDefault_minloc = True self.isDefault_majfmt = True self.isDefault_minfmt = True # This method is directly wrapped by Axes.set_{x,y}scale. def _set_axes_scale(self, value, **kwargs): """ Set this Axis' scale. Parameters ---------- value : {"linear", "log", "symlog", "logit", ...} or `.ScaleBase` The axis scale type to apply. **kwargs Different keyword arguments are accepted, depending on the scale. See the respective class keyword arguments: - `matplotlib.scale.LinearScale` - `matplotlib.scale.LogScale` - `matplotlib.scale.SymmetricalLogScale` - `matplotlib.scale.LogitScale` - `matplotlib.scale.FuncScale` - `matplotlib.scale.AsinhScale` Notes ----- By default, Matplotlib supports the above-mentioned scales. Additionally, custom scales may be registered using `matplotlib.scale.register_scale`. These scales can then also be used here. """ name = self._get_axis_name() old_default_lims = (self.get_major_locator() .nonsingular(-np.inf, np.inf)) for ax in self._get_shared_axes(): ax._axis_map[name]._set_scale(value, **kwargs) ax._update_transScale() ax.stale = True new_default_lims = (self.get_major_locator() .nonsingular(-np.inf, np.inf)) if old_default_lims != new_default_lims: # Force autoscaling now, to take advantage of the scale locator's # nonsingular() before it possibly gets swapped out by the user. self.axes.autoscale_view( **{f"scale{k}": k == name for k in self.axes._axis_names}) def limit_range_for_scale(self, vmin, vmax): return self._scale.limit_range_for_scale(vmin, vmax, self.get_minpos()) def _get_autoscale_on(self): """Return whether this Axis is autoscaled.""" return self._autoscale_on def _set_autoscale_on(self, b): """ Set whether this Axis is autoscaled when drawing or by `.Axes.autoscale_view`. If b is None, then the value is not changed. Parameters ---------- b : bool """ if b is not None: self._autoscale_on = b def get_children(self): return [self.label, self.offsetText, *self.get_major_ticks(), *self.get_minor_ticks()] def _reset_major_tick_kw(self): self._major_tick_kw.clear() self._major_tick_kw['gridOn'] = ( mpl.rcParams['axes.grid'] and mpl.rcParams['axes.grid.which'] in ('both', 'major')) def _reset_minor_tick_kw(self): self._minor_tick_kw.clear() self._minor_tick_kw['gridOn'] = ( mpl.rcParams['axes.grid'] and mpl.rcParams['axes.grid.which'] in ('both', 'minor')) def clear(self): """ Clear the axis. This resets axis properties to their default values: - the label - the scale - locators, formatters and ticks - major and minor grid - units - registered callbacks """ self.label._reset_visual_defaults() # The above resets the label formatting using text rcParams, # so we then update the formatting using axes rcParams self.label.set_color(mpl.rcParams['axes.labelcolor']) self.label.set_fontsize(mpl.rcParams['axes.labelsize']) self.label.set_fontweight(mpl.rcParams['axes.labelweight']) self.offsetText._reset_visual_defaults() self.labelpad = mpl.rcParams['axes.labelpad'] self._init() self._set_scale('linear') # Clear the callback registry for this axis, or it may "leak" self.callbacks = cbook.CallbackRegistry(signals=["units"]) # whether the grids are on self._major_tick_kw['gridOn'] = ( mpl.rcParams['axes.grid'] and mpl.rcParams['axes.grid.which'] in ('both', 'major')) self._minor_tick_kw['gridOn'] = ( mpl.rcParams['axes.grid'] and mpl.rcParams['axes.grid.which'] in ('both', 'minor')) self.reset_ticks() self.converter = None self.units = None self.stale = True def reset_ticks(self): """ Re-initialize the major and minor Tick lists. Each list starts with a single fresh Tick. """ # Restore the lazy tick lists. try: del self.majorTicks except AttributeError: pass try: del self.minorTicks except AttributeError: pass try: self.set_clip_path(self.axes.patch) except AttributeError: pass def minorticks_on(self): """ Display default minor ticks on the Axis, depending on the scale (`~.axis.Axis.get_scale`). Scales use specific minor locators: - log: `~.LogLocator` - symlog: `~.SymmetricalLogLocator` - asinh: `~.AsinhLocator` - logit: `~.LogitLocator` - default: `~.AutoMinorLocator` Displaying minor ticks may reduce performance; you may turn them off using `minorticks_off()` if drawing speed is a problem. """ scale = self.get_scale() if scale == 'log': s = self._scale self.set_minor_locator(mticker.LogLocator(s.base, s.subs)) elif scale == 'symlog': s = self._scale self.set_minor_locator( mticker.SymmetricalLogLocator(s._transform, s.subs)) elif scale == 'asinh': s = self._scale self.set_minor_locator( mticker.AsinhLocator(s.linear_width, base=s._base, subs=s._subs)) elif scale == 'logit': self.set_minor_locator(mticker.LogitLocator(minor=True)) else: self.set_minor_locator(mticker.AutoMinorLocator()) def minorticks_off(self): """Remove minor ticks from the Axis.""" self.set_minor_locator(mticker.NullLocator()) def set_tick_params(self, which='major', reset=False, **kwargs): """ Set appearance parameters for ticks, ticklabels, and gridlines. For documentation of keyword arguments, see :meth:`matplotlib.axes.Axes.tick_params`. See Also -------- .Axis.get_tick_params View the current style settings for ticks, ticklabels, and gridlines. """ _api.check_in_list(['major', 'minor', 'both'], which=which) kwtrans = self._translate_tick_params(kwargs) # the kwargs are stored in self._major/minor_tick_kw so that any # future new ticks will automatically get them if reset: if which in ['major', 'both']: self._reset_major_tick_kw() self._major_tick_kw.update(kwtrans) if which in ['minor', 'both']: self._reset_minor_tick_kw() self._minor_tick_kw.update(kwtrans) self.reset_ticks() else: if which in ['major', 'both']: self._major_tick_kw.update(kwtrans) for tick in self.majorTicks: tick._apply_params(**kwtrans) if which in ['minor', 'both']: self._minor_tick_kw.update(kwtrans) for tick in self.minorTicks: tick._apply_params(**kwtrans) # labelOn and labelcolor also apply to the offset text. if 'label1On' in kwtrans or 'label2On' in kwtrans: self.offsetText.set_visible( self._major_tick_kw.get('label1On', False) or self._major_tick_kw.get('label2On', False)) if 'labelcolor' in kwtrans: self.offsetText.set_color(kwtrans['labelcolor']) self.stale = True def get_tick_params(self, which='major'): """ Get appearance parameters for ticks, ticklabels, and gridlines. .. versionadded:: 3.7 Parameters ---------- which : {'major', 'minor'}, default: 'major' The group of ticks for which the parameters are retrieved. Returns ------- dict Properties for styling tick elements added to the axis. Notes ----- This method returns the appearance parameters for styling *new* elements added to this axis and may be different from the values on current elements if they were modified directly by the user (e.g., via ``set_*`` methods on individual tick objects). Examples -------- :: >>> ax.yaxis.set_tick_params(labelsize=30, labelcolor='red', ... direction='out', which='major') >>> ax.yaxis.get_tick_params(which='major') {'direction': 'out', 'left': True, 'right': False, 'labelleft': True, 'labelright': False, 'gridOn': False, 'labelsize': 30, 'labelcolor': 'red'} >>> ax.yaxis.get_tick_params(which='minor') {'left': True, 'right': False, 'labelleft': True, 'labelright': False, 'gridOn': False} """ _api.check_in_list(['major', 'minor'], which=which) if which == 'major': return self._translate_tick_params( self._major_tick_kw, reverse=True ) return self._translate_tick_params(self._minor_tick_kw, reverse=True) @staticmethod def _translate_tick_params(kw, reverse=False): """ Translate the kwargs supported by `.Axis.set_tick_params` to kwargs supported by `.Tick._apply_params`. In particular, this maps axis specific names like 'top', 'left' to the generic tick1, tick2 logic of the axis. Additionally, there are some other name translations. Returns a new dict of translated kwargs. Note: Use reverse=True to translate from those supported by `.Tick._apply_params` back to those supported by `.Axis.set_tick_params`. """ kw_ = {**kw} # The following lists may be moved to a more accessible location. allowed_keys = [ 'size', 'width', 'color', 'tickdir', 'pad', 'labelsize', 'labelcolor', 'labelfontfamily', 'zorder', 'gridOn', 'tick1On', 'tick2On', 'label1On', 'label2On', 'length', 'direction', 'left', 'bottom', 'right', 'top', 'labelleft', 'labelbottom', 'labelright', 'labeltop', 'labelrotation', *_gridline_param_names] keymap = { # tick_params key -> axis key 'length': 'size', 'direction': 'tickdir', 'rotation': 'labelrotation', 'left': 'tick1On', 'bottom': 'tick1On', 'right': 'tick2On', 'top': 'tick2On', 'labelleft': 'label1On', 'labelbottom': 'label1On', 'labelright': 'label2On', 'labeltop': 'label2On', } if reverse: kwtrans = { oldkey: kw_.pop(newkey) for oldkey, newkey in keymap.items() if newkey in kw_ } else: kwtrans = { newkey: kw_.pop(oldkey) for oldkey, newkey in keymap.items() if oldkey in kw_ } if 'colors' in kw_: c = kw_.pop('colors') kwtrans['color'] = c kwtrans['labelcolor'] = c # Maybe move the checking up to the caller of this method. for key in kw_: if key not in allowed_keys: raise ValueError( "keyword %s is not recognized; valid keywords are %s" % (key, allowed_keys)) kwtrans.update(kw_) return kwtrans @_api.rename_parameter("3.8", "clippath", "path") def set_clip_path(self, path, transform=None): super().set_clip_path(path, transform) for child in self.majorTicks + self.minorTicks: child.set_clip_path(path, transform) self.stale = True def get_view_interval(self): """Return the ``(min, max)`` view limits of this axis.""" raise NotImplementedError('Derived must override') def set_view_interval(self, vmin, vmax, ignore=False): """ Set the axis view limits. This method is for internal use; Matplotlib users should typically use e.g. `~.Axes.set_xlim` or `~.Axes.set_ylim`. If *ignore* is False (the default), this method will never reduce the preexisting view limits, only expand them if *vmin* or *vmax* are not within them. Moreover, the order of *vmin* and *vmax* does not matter; the orientation of the axis will not change. If *ignore* is True, the view limits will be set exactly to ``(vmin, vmax)`` in that order. """ raise NotImplementedError('Derived must override') def get_data_interval(self): """Return the ``(min, max)`` data limits of this axis.""" raise NotImplementedError('Derived must override') def set_data_interval(self, vmin, vmax, ignore=False): """ Set the axis data limits. This method is for internal use. If *ignore* is False (the default), this method will never reduce the preexisting data limits, only expand them if *vmin* or *vmax* are not within them. Moreover, the order of *vmin* and *vmax* does not matter; the orientation of the axis will not change. If *ignore* is True, the data limits will be set exactly to ``(vmin, vmax)`` in that order. """ raise NotImplementedError('Derived must override') def get_inverted(self): """ Return whether this Axis is oriented in the "inverse" direction. The "normal" direction is increasing to the right for the x-axis and to the top for the y-axis; the "inverse" direction is increasing to the left for the x-axis and to the bottom for the y-axis. """ low, high = self.get_view_interval() return high < low def set_inverted(self, inverted): """ Set whether this Axis is oriented in the "inverse" direction. The "normal" direction is increasing to the right for the x-axis and to the top for the y-axis; the "inverse" direction is increasing to the left for the x-axis and to the bottom for the y-axis. """ a, b = self.get_view_interval() # cast to bool to avoid bad interaction between python 3.8 and np.bool_ self._set_lim(*sorted((a, b), reverse=bool(inverted)), auto=None) def set_default_intervals(self): """ Set the default limits for the axis data and view interval if they have not been not mutated yet. """ # this is mainly in support of custom object plotting. For # example, if someone passes in a datetime object, we do not # know automagically how to set the default min/max of the # data and view limits. The unit conversion AxisInfo # interface provides a hook for custom types to register # default limits through the AxisInfo.default_limits # attribute, and the derived code below will check for that # and use it if it's available (else just use 0..1) def _set_lim(self, v0, v1, *, emit=True, auto): """ Set view limits. This method is a helper for the Axes ``set_xlim``, ``set_ylim``, and ``set_zlim`` methods. Parameters ---------- v0, v1 : float The view limits. (Passing *v0* as a (low, high) pair is not supported; normalization must occur in the Axes setters.) emit : bool, default: True Whether to notify observers of limit change. auto : bool or None, default: False Whether to turn on autoscaling of the x-axis. True turns on, False turns off, None leaves unchanged. """ name = self._get_axis_name() self.axes._process_unit_info([(name, (v0, v1))], convert=False) v0 = self.axes._validate_converted_limits(v0, self.convert_units) v1 = self.axes._validate_converted_limits(v1, self.convert_units) if v0 is None or v1 is None: # Axes init calls set_xlim(0, 1) before get_xlim() can be called, # so only grab the limits if we really need them. old0, old1 = self.get_view_interval() if v0 is None: v0 = old0 if v1 is None: v1 = old1 if self.get_scale() == 'log' and (v0 <= 0 or v1 <= 0): # Axes init calls set_xlim(0, 1) before get_xlim() can be called, # so only grab the limits if we really need them. old0, old1 = self.get_view_interval() if v0 <= 0: _api.warn_external(f"Attempt to set non-positive {name}lim on " f"a log-scaled axis will be ignored.") v0 = old0 if v1 <= 0: _api.warn_external(f"Attempt to set non-positive {name}lim on " f"a log-scaled axis will be ignored.") v1 = old1 if v0 == v1: _api.warn_external( f"Attempting to set identical low and high {name}lims " f"makes transformation singular; automatically expanding.") reverse = bool(v0 > v1) # explicit cast needed for python3.8+np.bool_. v0, v1 = self.get_major_locator().nonsingular(v0, v1) v0, v1 = self.limit_range_for_scale(v0, v1) v0, v1 = sorted([v0, v1], reverse=bool(reverse)) self.set_view_interval(v0, v1, ignore=True) # Mark viewlims as no longer stale without triggering an autoscale. for ax in self._get_shared_axes(): ax._stale_viewlims[name] = False self._set_autoscale_on(auto) if emit: self.axes.callbacks.process(f"{name}lim_changed", self.axes) # Call all of the other Axes that are shared with this one for other in self._get_shared_axes(): if other is self.axes: continue other._axis_map[name]._set_lim(v0, v1, emit=False, auto=auto) if emit: other.callbacks.process(f"{name}lim_changed", other) if other.figure != self.figure: other.figure.canvas.draw_idle() self.stale = True return v0, v1 def _set_artist_props(self, a): if a is None: return a.set_figure(self.figure) def _update_ticks(self): """ Update ticks (position and labels) using the current data interval of the axes. Return the list of ticks that will be drawn. """ major_locs = self.get_majorticklocs() major_labels = self.major.formatter.format_ticks(major_locs) major_ticks = self.get_major_ticks(len(major_locs)) for tick, loc, label in zip(major_ticks, major_locs, major_labels): tick.update_position(loc) tick.label1.set_text(label) tick.label2.set_text(label) minor_locs = self.get_minorticklocs() minor_labels = self.minor.formatter.format_ticks(minor_locs) minor_ticks = self.get_minor_ticks(len(minor_locs)) for tick, loc, label in zip(minor_ticks, minor_locs, minor_labels): tick.update_position(loc) tick.label1.set_text(label) tick.label2.set_text(label) ticks = [*major_ticks, *minor_ticks] view_low, view_high = self.get_view_interval() if view_low > view_high: view_low, view_high = view_high, view_low if (hasattr(self, "axes") and self.axes.name == '3d' and mpl.rcParams['axes3d.automargin']): # In mpl3.8, the margin was 1/48. Due to the change in automargin # behavior in mpl3.9, we need to adjust this to compensate for a # zoom factor of 2/48, giving us a 23/24 modifier. So the new # margin is 0.019965277777777776 = 1/48*23/24. margin = 0.019965277777777776 delta = view_high - view_low view_high = view_high - delta * margin view_low = view_low + delta * margin interval_t = self.get_transform().transform([view_low, view_high]) ticks_to_draw = [] for tick in ticks: try: loc_t = self.get_transform().transform(tick.get_loc()) except AssertionError: # transforms.transform doesn't allow masked values but # some scales might make them, so we need this try/except. pass else: if mtransforms._interval_contains_close(interval_t, loc_t): ticks_to_draw.append(tick) return ticks_to_draw def _get_ticklabel_bboxes(self, ticks, renderer=None): """Return lists of bboxes for ticks' label1's and label2's.""" if renderer is None: renderer = self.figure._get_renderer() return ([tick.label1.get_window_extent(renderer) for tick in ticks if tick.label1.get_visible()], [tick.label2.get_window_extent(renderer) for tick in ticks if tick.label2.get_visible()]) def get_tightbbox(self, renderer=None, *, for_layout_only=False): """ Return a bounding box that encloses the axis. It only accounts tick labels, axis label, and offsetText. If *for_layout_only* is True, then the width of the label (if this is an x-axis) or the height of the label (if this is a y-axis) is collapsed to near zero. This allows tight/constrained_layout to ignore too-long labels when doing their layout. """ if not self.get_visible(): return if renderer is None: renderer = self.figure._get_renderer() ticks_to_draw = self._update_ticks() self._update_label_position(renderer) # go back to just this axis's tick labels tlb1, tlb2 = self._get_ticklabel_bboxes(ticks_to_draw, renderer) self._update_offset_text_position(tlb1, tlb2) self.offsetText.set_text(self.major.formatter.get_offset()) bboxes = [ *(a.get_window_extent(renderer) for a in [self.offsetText] if a.get_visible()), *tlb1, *tlb2, ] # take care of label if self.label.get_visible(): bb = self.label.get_window_extent(renderer) # for constrained/tight_layout, we want to ignore the label's # width/height because the adjustments they make can't be improved. # this code collapses the relevant direction if for_layout_only: if self.axis_name == "x" and bb.width > 0: bb.x0 = (bb.x0 + bb.x1) / 2 - 0.5 bb.x1 = bb.x0 + 1.0 if self.axis_name == "y" and bb.height > 0: bb.y0 = (bb.y0 + bb.y1) / 2 - 0.5 bb.y1 = bb.y0 + 1.0 bboxes.append(bb) bboxes = [b for b in bboxes if 0 < b.width < np.inf and 0 < b.height < np.inf] if bboxes: return mtransforms.Bbox.union(bboxes) else: return None def get_tick_padding(self): values = [] if len(self.majorTicks): values.append(self.majorTicks[0].get_tick_padding()) if len(self.minorTicks): values.append(self.minorTicks[0].get_tick_padding()) return max(values, default=0) @martist.allow_rasterization def draw(self, renderer): # docstring inherited if not self.get_visible(): return renderer.open_group(__name__, gid=self.get_gid()) ticks_to_draw = self._update_ticks() tlb1, tlb2 = self._get_ticklabel_bboxes(ticks_to_draw, renderer) for tick in ticks_to_draw: tick.draw(renderer) # Shift label away from axes to avoid overlapping ticklabels. self._update_label_position(renderer) self.label.draw(renderer) self._update_offset_text_position(tlb1, tlb2) self.offsetText.set_text(self.major.formatter.get_offset()) self.offsetText.draw(renderer) renderer.close_group(__name__) self.stale = False def get_gridlines(self): r"""Return this Axis' grid lines as a list of `.Line2D`\s.""" ticks = self.get_major_ticks() return cbook.silent_list('Line2D gridline', [tick.gridline for tick in ticks]) def get_label(self): """Return the axis label as a Text instance.""" return self.label def get_offset_text(self): """Return the axis offsetText as a Text instance.""" return self.offsetText def get_pickradius(self): """Return the depth of the axis used by the picker.""" return self._pickradius def get_majorticklabels(self): """Return this Axis' major tick labels, as a list of `~.text.Text`.""" self._update_ticks() ticks = self.get_major_ticks() labels1 = [tick.label1 for tick in ticks if tick.label1.get_visible()] labels2 = [tick.label2 for tick in ticks if tick.label2.get_visible()] return labels1 + labels2 def get_minorticklabels(self): """Return this Axis' minor tick labels, as a list of `~.text.Text`.""" self._update_ticks() ticks = self.get_minor_ticks() labels1 = [tick.label1 for tick in ticks if tick.label1.get_visible()] labels2 = [tick.label2 for tick in ticks if tick.label2.get_visible()] return labels1 + labels2 def get_ticklabels(self, minor=False, which=None): """ Get this Axis' tick labels. Parameters ---------- minor : bool Whether to return the minor or the major ticklabels. which : None, ('minor', 'major', 'both') Overrides *minor*. Selects which ticklabels to return Returns ------- list of `~matplotlib.text.Text` """ if which is not None: if which == 'minor': return self.get_minorticklabels() elif which == 'major': return self.get_majorticklabels() elif which == 'both': return self.get_majorticklabels() + self.get_minorticklabels() else: _api.check_in_list(['major', 'minor', 'both'], which=which) if minor: return self.get_minorticklabels() return self.get_majorticklabels() def get_majorticklines(self): r"""Return this Axis' major tick lines as a list of `.Line2D`\s.""" lines = [] ticks = self.get_major_ticks() for tick in ticks: lines.append(tick.tick1line) lines.append(tick.tick2line) return cbook.silent_list('Line2D ticklines', lines) def get_minorticklines(self): r"""Return this Axis' minor tick lines as a list of `.Line2D`\s.""" lines = [] ticks = self.get_minor_ticks() for tick in ticks: lines.append(tick.tick1line) lines.append(tick.tick2line) return cbook.silent_list('Line2D ticklines', lines) def get_ticklines(self, minor=False): r"""Return this Axis' tick lines as a list of `.Line2D`\s.""" if minor: return self.get_minorticklines() return self.get_majorticklines() def get_majorticklocs(self): """Return this Axis' major tick locations in data coordinates.""" return self.major.locator() def get_minorticklocs(self): """Return this Axis' minor tick locations in data coordinates.""" # Remove minor ticks duplicating major ticks. minor_locs = np.asarray(self.minor.locator()) if self.remove_overlapping_locs: major_locs = self.major.locator() transform = self._scale.get_transform() tr_minor_locs = transform.transform(minor_locs) tr_major_locs = transform.transform(major_locs) lo, hi = sorted(transform.transform(self.get_view_interval())) # Use the transformed view limits as scale. 1e-5 is the default # rtol for np.isclose. tol = (hi - lo) * 1e-5 mask = np.isclose(tr_minor_locs[:, None], tr_major_locs[None, :], atol=tol, rtol=0).any(axis=1) minor_locs = minor_locs[~mask] return minor_locs def get_ticklocs(self, *, minor=False): """ Return this Axis' tick locations in data coordinates. The locations are not clipped to the current axis limits and hence may contain locations that are not visible in the output. Parameters ---------- minor : bool, default: False True to return the minor tick directions, False to return the major tick directions. Returns ------- array of tick locations """ return self.get_minorticklocs() if minor else self.get_majorticklocs() def get_ticks_direction(self, minor=False): """ Return an array of this Axis' tick directions. Parameters ---------- minor : bool, default: False True to return the minor tick directions, False to return the major tick directions. Returns ------- array of tick directions """ if minor: return np.array( [tick._tickdir for tick in self.get_minor_ticks()]) else: return np.array( [tick._tickdir for tick in self.get_major_ticks()]) def _get_tick(self, major): """Return the default tick instance.""" if self._tick_class is None: raise NotImplementedError( f"The Axis subclass {self.__class__.__name__} must define " "_tick_class or reimplement _get_tick()") tick_kw = self._major_tick_kw if major else self._minor_tick_kw return self._tick_class(self.axes, 0, major=major, **tick_kw) def _get_tick_label_size(self, axis_name): """ Return the text size of tick labels for this Axis. This is a convenience function to avoid having to create a `Tick` in `.get_tick_space`, since it is expensive. """ tick_kw = self._major_tick_kw size = tick_kw.get('labelsize', mpl.rcParams[f'{axis_name}tick.labelsize']) return mtext.FontProperties(size=size).get_size_in_points() def _copy_tick_props(self, src, dest): """Copy the properties from *src* tick to *dest* tick.""" if src is None or dest is None: return dest.label1.update_from(src.label1) dest.label2.update_from(src.label2) dest.tick1line.update_from(src.tick1line) dest.tick2line.update_from(src.tick2line) dest.gridline.update_from(src.gridline) dest.update_from(src) dest._loc = src._loc dest._size = src._size dest._width = src._width dest._base_pad = src._base_pad dest._labelrotation = src._labelrotation dest._zorder = src._zorder dest._tickdir = src._tickdir def get_label_text(self): """Get the text of the label.""" return self.label.get_text() def get_major_locator(self): """Get the locator of the major ticker.""" return self.major.locator def get_minor_locator(self): """Get the locator of the minor ticker.""" return self.minor.locator def get_major_formatter(self): """Get the formatter of the major ticker.""" return self.major.formatter def get_minor_formatter(self): """Get the formatter of the minor ticker.""" return self.minor.formatter def get_major_ticks(self, numticks=None): r""" Return the list of major `.Tick`\s. .. warning:: Ticks are not guaranteed to be persistent. Various operations can create, delete and modify the Tick instances. There is an imminent risk that changes to individual ticks will not survive if you work on the figure further (including also panning/zooming on a displayed figure). Working on the individual ticks is a method of last resort. Use `.set_tick_params` instead if possible. """ if numticks is None: numticks = len(self.get_majorticklocs()) while len(self.majorTicks) < numticks: # Update the new tick label properties from the old. tick = self._get_tick(major=True) self.majorTicks.append(tick) self._copy_tick_props(self.majorTicks[0], tick) return self.majorTicks[:numticks] def get_minor_ticks(self, numticks=None): r""" Return the list of minor `.Tick`\s. .. warning:: Ticks are not guaranteed to be persistent. Various operations can create, delete and modify the Tick instances. There is an imminent risk that changes to individual ticks will not survive if you work on the figure further (including also panning/zooming on a displayed figure). Working on the individual ticks is a method of last resort. Use `.set_tick_params` instead if possible. """ if numticks is None: numticks = len(self.get_minorticklocs()) while len(self.minorTicks) < numticks: # Update the new tick label properties from the old. tick = self._get_tick(major=False) self.minorTicks.append(tick) self._copy_tick_props(self.minorTicks[0], tick) return self.minorTicks[:numticks] def grid(self, visible=None, which='major', **kwargs): """ Configure the grid lines. Parameters ---------- visible : bool or None Whether to show the grid lines. If any *kwargs* are supplied, it is assumed you want the grid on and *visible* will be set to True. If *visible* is *None* and there are no *kwargs*, this toggles the visibility of the lines. which : {'major', 'minor', 'both'} The grid lines to apply the changes on. **kwargs : `~matplotlib.lines.Line2D` properties Define the line properties of the grid, e.g.:: grid(color='r', linestyle='-', linewidth=2) """ if kwargs: if visible is None: visible = True elif not visible: # something false-like but not None _api.warn_external('First parameter to grid() is false, ' 'but line properties are supplied. The ' 'grid will be enabled.') visible = True which = which.lower() _api.check_in_list(['major', 'minor', 'both'], which=which) gridkw = {f'grid_{name}': value for name, value in kwargs.items()} if which in ['minor', 'both']: gridkw['gridOn'] = (not self._minor_tick_kw['gridOn'] if visible is None else visible) self.set_tick_params(which='minor', **gridkw) if which in ['major', 'both']: gridkw['gridOn'] = (not self._major_tick_kw['gridOn'] if visible is None else visible) self.set_tick_params(which='major', **gridkw) self.stale = True def update_units(self, data): """ Introspect *data* for units converter and update the ``axis.converter`` instance if necessary. Return *True* if *data* is registered for unit conversion. """ converter = munits.registry.get_converter(data) if converter is None: return False neednew = self.converter != converter self.converter = converter default = self.converter.default_units(data, self) if default is not None and self.units is None: self.set_units(default) elif neednew: self._update_axisinfo() self.stale = True return True def _update_axisinfo(self): """ Check the axis converter for the stored units to see if the axis info needs to be updated. """ if self.converter is None: return info = self.converter.axisinfo(self.units, self) if info is None: return if info.majloc is not None and \ self.major.locator != info.majloc and self.isDefault_majloc: self.set_major_locator(info.majloc) self.isDefault_majloc = True if info.minloc is not None and \ self.minor.locator != info.minloc and self.isDefault_minloc: self.set_minor_locator(info.minloc) self.isDefault_minloc = True if info.majfmt is not None and \ self.major.formatter != info.majfmt and self.isDefault_majfmt: self.set_major_formatter(info.majfmt) self.isDefault_majfmt = True if info.minfmt is not None and \ self.minor.formatter != info.minfmt and self.isDefault_minfmt: self.set_minor_formatter(info.minfmt) self.isDefault_minfmt = True if info.label is not None and self.isDefault_label: self.set_label_text(info.label) self.isDefault_label = True self.set_default_intervals() def have_units(self): return self.converter is not None or self.units is not None def convert_units(self, x): # If x is natively supported by Matplotlib, doesn't need converting if munits._is_natively_supported(x): return x if self.converter is None: self.converter = munits.registry.get_converter(x) if self.converter is None: return x try: ret = self.converter.convert(x, self.units, self) except Exception as e: raise munits.ConversionError('Failed to convert value(s) to axis ' f'units: {x!r}') from e return ret def set_units(self, u): """ Set the units for axis. Parameters ---------- u : units tag Notes ----- The units of any shared axis will also be updated. """ if u == self.units: return for axis in self._get_shared_axis(): axis.units = u axis._update_axisinfo() axis.callbacks.process('units') axis.stale = True def get_units(self): """Return the units for axis.""" return self.units def set_label_text(self, label, fontdict=None, **kwargs): """ Set the text value of the axis label. Parameters ---------- label : str Text string. fontdict : dict Text properties. .. admonition:: Discouraged The use of *fontdict* is discouraged. Parameters should be passed as individual keyword arguments or using dictionary-unpacking ``set_label_text(..., **fontdict)``. **kwargs Merged into fontdict. """ self.isDefault_label = False self.label.set_text(label) if fontdict is not None: self.label.update(fontdict) self.label.update(kwargs) self.stale = True return self.label def set_major_formatter(self, formatter): """ Set the formatter of the major ticker. In addition to a `~matplotlib.ticker.Formatter` instance, this also accepts a ``str`` or function. For a ``str`` a `~matplotlib.ticker.StrMethodFormatter` is used. The field used for the value must be labeled ``'x'`` and the field used for the position must be labeled ``'pos'``. See the `~matplotlib.ticker.StrMethodFormatter` documentation for more information. For a function, a `~matplotlib.ticker.FuncFormatter` is used. The function must take two inputs (a tick value ``x`` and a position ``pos``), and return a string containing the corresponding tick label. See the `~matplotlib.ticker.FuncFormatter` documentation for more information. Parameters ---------- formatter : `~matplotlib.ticker.Formatter`, ``str``, or function """ self._set_formatter(formatter, self.major) def set_minor_formatter(self, formatter): """ Set the formatter of the minor ticker. In addition to a `~matplotlib.ticker.Formatter` instance, this also accepts a ``str`` or function. See `.Axis.set_major_formatter` for more information. Parameters ---------- formatter : `~matplotlib.ticker.Formatter`, ``str``, or function """ self._set_formatter(formatter, self.minor) def _set_formatter(self, formatter, level): if isinstance(formatter, str): formatter = mticker.StrMethodFormatter(formatter) # Don't allow any other TickHelper to avoid easy-to-make errors, # like using a Locator instead of a Formatter. elif (callable(formatter) and not isinstance(formatter, mticker.TickHelper)): formatter = mticker.FuncFormatter(formatter) else: _api.check_isinstance(mticker.Formatter, formatter=formatter) if (isinstance(formatter, mticker.FixedFormatter) and len(formatter.seq) > 0 and not isinstance(level.locator, mticker.FixedLocator)): _api.warn_external('FixedFormatter should only be used together ' 'with FixedLocator') if level == self.major: self.isDefault_majfmt = False else: self.isDefault_minfmt = False level.formatter = formatter formatter.set_axis(self) self.stale = True def set_major_locator(self, locator): """ Set the locator of the major ticker. Parameters ---------- locator : `~matplotlib.ticker.Locator` """ _api.check_isinstance(mticker.Locator, locator=locator) self.isDefault_majloc = False self.major.locator = locator if self.major.formatter: self.major.formatter._set_locator(locator) locator.set_axis(self) self.stale = True def set_minor_locator(self, locator): """ Set the locator of the minor ticker. Parameters ---------- locator : `~matplotlib.ticker.Locator` """ _api.check_isinstance(mticker.Locator, locator=locator) self.isDefault_minloc = False self.minor.locator = locator if self.minor.formatter: self.minor.formatter._set_locator(locator) locator.set_axis(self) self.stale = True def set_pickradius(self, pickradius): """ Set the depth of the axis used by the picker. Parameters ---------- pickradius : float The acceptance radius for containment tests. See also `.Axis.contains`. """ if not isinstance(pickradius, Real) or pickradius < 0: raise ValueError("pick radius should be a distance") self._pickradius = pickradius pickradius = property( get_pickradius, set_pickradius, doc="The acceptance radius for " "containment tests. See also `.Axis.contains`.") # Helper for set_ticklabels. Defining it here makes it picklable. @staticmethod def _format_with_dict(tickd, x, pos): return tickd.get(x, "") def set_ticklabels(self, labels, *, minor=False, fontdict=None, **kwargs): r""" [*Discouraged*] Set this Axis' tick labels with list of string labels. .. admonition:: Discouraged The use of this method is discouraged, because of the dependency on tick positions. In most cases, you'll want to use ``Axes.set_[x/y/z]ticks(positions, labels)`` or ``Axis.set_ticks`` instead. If you are using this method, you should always fix the tick positions before, e.g. by using `.Axis.set_ticks` or by explicitly setting a `~.ticker.FixedLocator`. Otherwise, ticks are free to move and the labels may end up in unexpected positions. Parameters ---------- labels : sequence of str or of `.Text`\s Texts for labeling each tick location in the sequence set by `.Axis.set_ticks`; the number of labels must match the number of locations. The labels are used as is, via a `.FixedFormatter` (without further formatting). minor : bool If True, set minor ticks instead of major ticks. fontdict : dict, optional .. admonition:: Discouraged The use of *fontdict* is discouraged. Parameters should be passed as individual keyword arguments or using dictionary-unpacking ``set_ticklabels(..., **fontdict)``. A dictionary controlling the appearance of the ticklabels. The default *fontdict* is:: {'fontsize': rcParams['axes.titlesize'], 'fontweight': rcParams['axes.titleweight'], 'verticalalignment': 'baseline', 'horizontalalignment': loc} **kwargs Text properties. .. warning:: This only sets the properties of the current ticks, which is only sufficient for static plots. Ticks are not guaranteed to be persistent. Various operations can create, delete and modify the Tick instances. There is an imminent risk that these settings can get lost if you work on the figure further (including also panning/zooming on a displayed figure). Use `.set_tick_params` instead if possible. Returns ------- list of `.Text`\s For each tick, includes ``tick.label1`` if it is visible, then ``tick.label2`` if it is visible, in that order. """ try: labels = [t.get_text() if hasattr(t, 'get_text') else t for t in labels] except TypeError: raise TypeError(f"{labels:=} must be a sequence") from None locator = (self.get_minor_locator() if minor else self.get_major_locator()) if not labels: # eg labels=[]: formatter = mticker.NullFormatter() elif isinstance(locator, mticker.FixedLocator): # Passing [] as a list of labels is often used as a way to # remove all tick labels, so only error for > 0 labels if len(locator.locs) != len(labels) and len(labels) != 0: raise ValueError( "The number of FixedLocator locations" f" ({len(locator.locs)}), usually from a call to" " set_ticks, does not match" f" the number of labels ({len(labels)}).") tickd = {loc: lab for loc, lab in zip(locator.locs, labels)} func = functools.partial(self._format_with_dict, tickd) formatter = mticker.FuncFormatter(func) else: _api.warn_external( "set_ticklabels() should only be used with a fixed number of " "ticks, i.e. after set_ticks() or using a FixedLocator.") formatter = mticker.FixedFormatter(labels) with warnings.catch_warnings(): warnings.filterwarnings( "ignore", message="FixedFormatter should only be used together with FixedLocator") if minor: self.set_minor_formatter(formatter) locs = self.get_minorticklocs() ticks = self.get_minor_ticks(len(locs)) else: self.set_major_formatter(formatter) locs = self.get_majorticklocs() ticks = self.get_major_ticks(len(locs)) ret = [] if fontdict is not None: kwargs.update(fontdict) for pos, (loc, tick) in enumerate(zip(locs, ticks)): tick.update_position(loc) tick_label = formatter(loc, pos) # deal with label1 tick.label1.set_text(tick_label) tick.label1._internal_update(kwargs) # deal with label2 tick.label2.set_text(tick_label) tick.label2._internal_update(kwargs) # only return visible tick labels if tick.label1.get_visible(): ret.append(tick.label1) if tick.label2.get_visible(): ret.append(tick.label2) self.stale = True return ret def _set_tick_locations(self, ticks, *, minor=False): # see docstring of set_ticks # XXX if the user changes units, the information will be lost here ticks = self.convert_units(ticks) locator = mticker.FixedLocator(ticks) # validate ticks early. if len(ticks): for axis in self._get_shared_axis(): # set_view_interval maintains any preexisting inversion. axis.set_view_interval(min(ticks), max(ticks)) self.axes.stale = True if minor: self.set_minor_locator(locator) return self.get_minor_ticks(len(ticks)) else: self.set_major_locator(locator) return self.get_major_ticks(len(ticks)) def set_ticks(self, ticks, labels=None, *, minor=False, **kwargs): """ Set this Axis' tick locations and optionally tick labels. If necessary, the view limits of the Axis are expanded so that all given ticks are visible. Parameters ---------- ticks : 1D array-like Array of tick locations (either floats or in axis units). The axis `.Locator` is replaced by a `~.ticker.FixedLocator`. Pass an empty list (``set_ticks([])``) to remove all ticks. Some tick formatters will not label arbitrary tick positions; e.g. log formatters only label decade ticks by default. In such a case you can set a formatter explicitly on the axis using `.Axis.set_major_formatter` or provide formatted *labels* yourself. labels : list of str, optional Tick labels for each location in *ticks*; must have the same length as *ticks*. If set, the labels are used as is, via a `.FixedFormatter`. If not set, the labels are generated using the axis tick `.Formatter`. minor : bool, default: False If ``False``, set only the major ticks; if ``True``, only the minor ticks. **kwargs `.Text` properties for the labels. Using these is only allowed if you pass *labels*. In other cases, please use `~.Axes.tick_params`. Notes ----- The mandatory expansion of the view limits is an intentional design choice to prevent the surprise of a non-visible tick. If you need other limits, you should set the limits explicitly after setting the ticks. """ if labels is None and kwargs: first_key = next(iter(kwargs)) raise ValueError( f"Incorrect use of keyword argument {first_key!r}. Keyword arguments " "other than 'minor' modify the text labels and can only be used if " "'labels' are passed as well.") result = self._set_tick_locations(ticks, minor=minor) if labels is not None: self.set_ticklabels(labels, minor=minor, **kwargs) return result def _get_tick_boxes_siblings(self, renderer): """ Get the bounding boxes for this `.axis` and its siblings as set by `.Figure.align_xlabels` or `.Figure.align_ylabels`. By default, it just gets bboxes for *self*. """ # Get the Grouper keeping track of x or y label groups for this figure. name = self._get_axis_name() if name not in self.figure._align_label_groups: return [], [] grouper = self.figure._align_label_groups[name] bboxes = [] bboxes2 = [] # If we want to align labels from other Axes: for ax in grouper.get_siblings(self.axes): axis = ax._axis_map[name] ticks_to_draw = axis._update_ticks() tlb, tlb2 = axis._get_ticklabel_bboxes(ticks_to_draw, renderer) bboxes.extend(tlb) bboxes2.extend(tlb2) return bboxes, bboxes2 def _update_label_position(self, renderer): """ Update the label position based on the bounding box enclosing all the ticklabels and axis spine. """ raise NotImplementedError('Derived must override') def _update_offset_text_position(self, bboxes, bboxes2): """ Update the offset text position based on the sequence of bounding boxes of all the ticklabels. """ raise NotImplementedError('Derived must override') def axis_date(self, tz=None): """ Set up axis ticks and labels to treat data along this Axis as dates. Parameters ---------- tz : str or `datetime.tzinfo`, default: :rc:`timezone` The timezone used to create date labels. """ # By providing a sample datetime instance with the desired timezone, # the registered converter can be selected, and the "units" attribute, # which is the timezone, can be set. if isinstance(tz, str): import dateutil.tz tz = dateutil.tz.gettz(tz) self.update_units(datetime.datetime(2009, 1, 1, 0, 0, 0, 0, tz)) def get_tick_space(self): """Return the estimated number of ticks that can fit on the axis.""" # Must be overridden in the subclass raise NotImplementedError() def _get_ticks_position(self): """ Helper for `XAxis.get_ticks_position` and `YAxis.get_ticks_position`. Check the visibility of tick1line, label1, tick2line, and label2 on the first major and the first minor ticks, and return - 1 if only tick1line and label1 are visible (which corresponds to "bottom" for the x-axis and "left" for the y-axis); - 2 if only tick2line and label2 are visible (which corresponds to "top" for the x-axis and "right" for the y-axis); - "default" if only tick1line, tick2line and label1 are visible; - "unknown" otherwise. """ major = self.majorTicks[0] minor = self.minorTicks[0] if all(tick.tick1line.get_visible() and not tick.tick2line.get_visible() and tick.label1.get_visible() and not tick.label2.get_visible() for tick in [major, minor]): return 1 elif all(tick.tick2line.get_visible() and not tick.tick1line.get_visible() and tick.label2.get_visible() and not tick.label1.get_visible() for tick in [major, minor]): return 2 elif all(tick.tick1line.get_visible() and tick.tick2line.get_visible() and tick.label1.get_visible() and not tick.label2.get_visible() for tick in [major, minor]): return "default" else: return "unknown" def get_label_position(self): """ Return the label position (top or bottom) """ return self.label_position def set_label_position(self, position): """ Set the label position (top or bottom) Parameters ---------- position : {'top', 'bottom'} """ raise NotImplementedError() def get_minpos(self): raise NotImplementedError() def _make_getset_interval(method_name, lim_name, attr_name): """ Helper to generate ``get_{data,view}_interval`` and ``set_{data,view}_interval`` implementations. """ def getter(self): # docstring inherited. return getattr(getattr(self.axes, lim_name), attr_name) def setter(self, vmin, vmax, ignore=False): # docstring inherited. if ignore: setattr(getattr(self.axes, lim_name), attr_name, (vmin, vmax)) else: oldmin, oldmax = getter(self) if oldmin < oldmax: setter(self, min(vmin, vmax, oldmin), max(vmin, vmax, oldmax), ignore=True) else: setter(self, max(vmin, vmax, oldmin), min(vmin, vmax, oldmax), ignore=True) self.stale = True getter.__name__ = f"get_{method_name}_interval" setter.__name__ = f"set_{method_name}_interval" return getter, setter class XAxis(Axis): __name__ = 'xaxis' axis_name = 'x' #: Read-only name identifying the axis. _tick_class = XTick def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self._init() def _init(self): """ Initialize the label and offsetText instance values and `label_position` / `offset_text_position`. """ # x in axes coords, y in display coords (to be updated at draw time by # _update_label_positions and _update_offset_text_position). self.label.set( x=0.5, y=0, verticalalignment='top', horizontalalignment='center', transform=mtransforms.blended_transform_factory( self.axes.transAxes, mtransforms.IdentityTransform()), ) self.label_position = 'bottom' if mpl.rcParams['xtick.labelcolor'] == 'inherit': tick_color = mpl.rcParams['xtick.color'] else: tick_color = mpl.rcParams['xtick.labelcolor'] self.offsetText.set( x=1, y=0, verticalalignment='top', horizontalalignment='right', transform=mtransforms.blended_transform_factory( self.axes.transAxes, mtransforms.IdentityTransform()), fontsize=mpl.rcParams['xtick.labelsize'], color=tick_color ) self.offset_text_position = 'bottom' def contains(self, mouseevent): """Test whether the mouse event occurred in the x-axis.""" if self._different_canvas(mouseevent): return False, {} x, y = mouseevent.x, mouseevent.y try: trans = self.axes.transAxes.inverted() xaxes, yaxes = trans.transform((x, y)) except ValueError: return False, {} (l, b), (r, t) = self.axes.transAxes.transform([(0, 0), (1, 1)]) inaxis = 0 <= xaxes <= 1 and ( b - self._pickradius < y < b or t < y < t + self._pickradius) return inaxis, {} def set_label_position(self, position): """ Set the label position (top or bottom) Parameters ---------- position : {'top', 'bottom'} """ self.label.set_verticalalignment(_api.check_getitem({ 'top': 'baseline', 'bottom': 'top', }, position=position)) self.label_position = position self.stale = True def _update_label_position(self, renderer): """ Update the label position based on the bounding box enclosing all the ticklabels and axis spine """ if not self._autolabelpos: return # get bounding boxes for this axis and any siblings # that have been set by `fig.align_xlabels()` bboxes, bboxes2 = self._get_tick_boxes_siblings(renderer=renderer) x, y = self.label.get_position() if self.label_position == 'bottom': try: spine = self.axes.spines['bottom'] spinebbox = spine.get_window_extent() except KeyError: # use Axes if spine doesn't exist spinebbox = self.axes.bbox bbox = mtransforms.Bbox.union(bboxes + [spinebbox]) bottom = bbox.y0 self.label.set_position( (x, bottom - self.labelpad * self.figure.dpi / 72) ) else: try: spine = self.axes.spines['top'] spinebbox = spine.get_window_extent() except KeyError: # use Axes if spine doesn't exist spinebbox = self.axes.bbox bbox = mtransforms.Bbox.union(bboxes2 + [spinebbox]) top = bbox.y1 self.label.set_position( (x, top + self.labelpad * self.figure.dpi / 72) ) def _update_offset_text_position(self, bboxes, bboxes2): """ Update the offset_text position based on the sequence of bounding boxes of all the ticklabels """ x, y = self.offsetText.get_position() if not hasattr(self, '_tick_position'): self._tick_position = 'bottom' if self._tick_position == 'bottom': if not len(bboxes): bottom = self.axes.bbox.ymin else: bbox = mtransforms.Bbox.union(bboxes) bottom = bbox.y0 y = bottom - self.OFFSETTEXTPAD * self.figure.dpi / 72 else: if not len(bboxes2): top = self.axes.bbox.ymax else: bbox = mtransforms.Bbox.union(bboxes2) top = bbox.y1 y = top + self.OFFSETTEXTPAD * self.figure.dpi / 72 self.offsetText.set_position((x, y)) def set_ticks_position(self, position): """ Set the ticks position. Parameters ---------- position : {'top', 'bottom', 'both', 'default', 'none'} 'both' sets the ticks to appear on both positions, but does not change the tick labels. 'default' resets the tick positions to the default: ticks on both positions, labels at bottom. 'none' can be used if you don't want any ticks. 'none' and 'both' affect only the ticks, not the labels. """ if position == 'top': self.set_tick_params(which='both', top=True, labeltop=True, bottom=False, labelbottom=False) self._tick_position = 'top' self.offsetText.set_verticalalignment('bottom') elif position == 'bottom': self.set_tick_params(which='both', top=False, labeltop=False, bottom=True, labelbottom=True) self._tick_position = 'bottom' self.offsetText.set_verticalalignment('top') elif position == 'both': self.set_tick_params(which='both', top=True, bottom=True) elif position == 'none': self.set_tick_params(which='both', top=False, bottom=False) elif position == 'default': self.set_tick_params(which='both', top=True, labeltop=False, bottom=True, labelbottom=True) self._tick_position = 'bottom' self.offsetText.set_verticalalignment('top') else: _api.check_in_list(['top', 'bottom', 'both', 'default', 'none'], position=position) self.stale = True def tick_top(self): """ Move ticks and ticklabels (if present) to the top of the Axes. """ label = True if 'label1On' in self._major_tick_kw: label = (self._major_tick_kw['label1On'] or self._major_tick_kw['label2On']) self.set_ticks_position('top') # If labels were turned off before this was called, leave them off. self.set_tick_params(which='both', labeltop=label) def tick_bottom(self): """ Move ticks and ticklabels (if present) to the bottom of the Axes. """ label = True if 'label1On' in self._major_tick_kw: label = (self._major_tick_kw['label1On'] or self._major_tick_kw['label2On']) self.set_ticks_position('bottom') # If labels were turned off before this was called, leave them off. self.set_tick_params(which='both', labelbottom=label) def get_ticks_position(self): """ Return the ticks position ("top", "bottom", "default", or "unknown"). """ return {1: "bottom", 2: "top", "default": "default", "unknown": "unknown"}[ self._get_ticks_position()] get_view_interval, set_view_interval = _make_getset_interval( "view", "viewLim", "intervalx") get_data_interval, set_data_interval = _make_getset_interval( "data", "dataLim", "intervalx") def get_minpos(self): return self.axes.dataLim.minposx def set_default_intervals(self): # docstring inherited # only change view if dataLim has not changed and user has # not changed the view: if (not self.axes.dataLim.mutatedx() and not self.axes.viewLim.mutatedx()): if self.converter is not None: info = self.converter.axisinfo(self.units, self) if info.default_limits is not None: xmin, xmax = self.convert_units(info.default_limits) self.axes.viewLim.intervalx = xmin, xmax self.stale = True def get_tick_space(self): ends = mtransforms.Bbox.unit().transformed( self.axes.transAxes - self.figure.dpi_scale_trans) length = ends.width * 72 # There is a heuristic here that the aspect ratio of tick text # is no more than 3:1 size = self._get_tick_label_size('x') * 3 if size > 0: return int(np.floor(length / size)) else: return 2**31 - 1 class YAxis(Axis): __name__ = 'yaxis' axis_name = 'y' #: Read-only name identifying the axis. _tick_class = YTick def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self._init() def _init(self): """ Initialize the label and offsetText instance values and `label_position` / `offset_text_position`. """ # x in display coords, y in axes coords (to be updated at draw time by # _update_label_positions and _update_offset_text_position). self.label.set( x=0, y=0.5, verticalalignment='bottom', horizontalalignment='center', rotation='vertical', rotation_mode='anchor', transform=mtransforms.blended_transform_factory( mtransforms.IdentityTransform(), self.axes.transAxes), ) self.label_position = 'left' if mpl.rcParams['ytick.labelcolor'] == 'inherit': tick_color = mpl.rcParams['ytick.color'] else: tick_color = mpl.rcParams['ytick.labelcolor'] # x in axes coords, y in display coords(!). self.offsetText.set( x=0, y=0.5, verticalalignment='baseline', horizontalalignment='left', transform=mtransforms.blended_transform_factory( self.axes.transAxes, mtransforms.IdentityTransform()), fontsize=mpl.rcParams['ytick.labelsize'], color=tick_color ) self.offset_text_position = 'left' def contains(self, mouseevent): # docstring inherited if self._different_canvas(mouseevent): return False, {} x, y = mouseevent.x, mouseevent.y try: trans = self.axes.transAxes.inverted() xaxes, yaxes = trans.transform((x, y)) except ValueError: return False, {} (l, b), (r, t) = self.axes.transAxes.transform([(0, 0), (1, 1)]) inaxis = 0 <= yaxes <= 1 and ( l - self._pickradius < x < l or r < x < r + self._pickradius) return inaxis, {} def set_label_position(self, position): """ Set the label position (left or right) Parameters ---------- position : {'left', 'right'} """ self.label.set_rotation_mode('anchor') self.label.set_verticalalignment(_api.check_getitem({ 'left': 'bottom', 'right': 'top', }, position=position)) self.label_position = position self.stale = True def _update_label_position(self, renderer): """ Update the label position based on the bounding box enclosing all the ticklabels and axis spine """ if not self._autolabelpos: return # get bounding boxes for this axis and any siblings # that have been set by `fig.align_ylabels()` bboxes, bboxes2 = self._get_tick_boxes_siblings(renderer=renderer) x, y = self.label.get_position() if self.label_position == 'left': try: spine = self.axes.spines['left'] spinebbox = spine.get_window_extent() except KeyError: # use Axes if spine doesn't exist spinebbox = self.axes.bbox bbox = mtransforms.Bbox.union(bboxes + [spinebbox]) left = bbox.x0 self.label.set_position( (left - self.labelpad * self.figure.dpi / 72, y) ) else: try: spine = self.axes.spines['right'] spinebbox = spine.get_window_extent() except KeyError: # use Axes if spine doesn't exist spinebbox = self.axes.bbox bbox = mtransforms.Bbox.union(bboxes2 + [spinebbox]) right = bbox.x1 self.label.set_position( (right + self.labelpad * self.figure.dpi / 72, y) ) def _update_offset_text_position(self, bboxes, bboxes2): """ Update the offset_text position based on the sequence of bounding boxes of all the ticklabels """ x, _ = self.offsetText.get_position() if 'outline' in self.axes.spines: # Special case for colorbars: bbox = self.axes.spines['outline'].get_window_extent() else: bbox = self.axes.bbox top = bbox.ymax self.offsetText.set_position( (x, top + self.OFFSETTEXTPAD * self.figure.dpi / 72) ) def set_offset_position(self, position): """ Parameters ---------- position : {'left', 'right'} """ x, y = self.offsetText.get_position() x = _api.check_getitem({'left': 0, 'right': 1}, position=position) self.offsetText.set_ha(position) self.offsetText.set_position((x, y)) self.stale = True def set_ticks_position(self, position): """ Set the ticks position. Parameters ---------- position : {'left', 'right', 'both', 'default', 'none'} 'both' sets the ticks to appear on both positions, but does not change the tick labels. 'default' resets the tick positions to the default: ticks on both positions, labels at left. 'none' can be used if you don't want any ticks. 'none' and 'both' affect only the ticks, not the labels. """ if position == 'right': self.set_tick_params(which='both', right=True, labelright=True, left=False, labelleft=False) self.set_offset_position(position) elif position == 'left': self.set_tick_params(which='both', right=False, labelright=False, left=True, labelleft=True) self.set_offset_position(position) elif position == 'both': self.set_tick_params(which='both', right=True, left=True) elif position == 'none': self.set_tick_params(which='both', right=False, left=False) elif position == 'default': self.set_tick_params(which='both', right=True, labelright=False, left=True, labelleft=True) else: _api.check_in_list(['left', 'right', 'both', 'default', 'none'], position=position) self.stale = True def tick_right(self): """ Move ticks and ticklabels (if present) to the right of the Axes. """ label = True if 'label1On' in self._major_tick_kw: label = (self._major_tick_kw['label1On'] or self._major_tick_kw['label2On']) self.set_ticks_position('right') # if labels were turned off before this was called # leave them off self.set_tick_params(which='both', labelright=label) def tick_left(self): """ Move ticks and ticklabels (if present) to the left of the Axes. """ label = True if 'label1On' in self._major_tick_kw: label = (self._major_tick_kw['label1On'] or self._major_tick_kw['label2On']) self.set_ticks_position('left') # if labels were turned off before this was called # leave them off self.set_tick_params(which='both', labelleft=label) def get_ticks_position(self): """ Return the ticks position ("left", "right", "default", or "unknown"). """ return {1: "left", 2: "right", "default": "default", "unknown": "unknown"}[ self._get_ticks_position()] get_view_interval, set_view_interval = _make_getset_interval( "view", "viewLim", "intervaly") get_data_interval, set_data_interval = _make_getset_interval( "data", "dataLim", "intervaly") def get_minpos(self): return self.axes.dataLim.minposy def set_default_intervals(self): # docstring inherited # only change view if dataLim has not changed and user has # not changed the view: if (not self.axes.dataLim.mutatedy() and not self.axes.viewLim.mutatedy()): if self.converter is not None: info = self.converter.axisinfo(self.units, self) if info.default_limits is not None: ymin, ymax = self.convert_units(info.default_limits) self.axes.viewLim.intervaly = ymin, ymax self.stale = True def get_tick_space(self): ends = mtransforms.Bbox.unit().transformed( self.axes.transAxes - self.figure.dpi_scale_trans) length = ends.height * 72 # Having a spacing of at least 2 just looks good. size = self._get_tick_label_size('y') * 2 if size > 0: return int(np.floor(length / size)) else: return 2**31 - 1