phyks/replot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Enable the use of orthonormal axis.

Browse Source 
Also move the `_render` method to the public API and add `xlim` and
`ylim` extra keywords arguments.

Closes https://github.com/Phyks/replot/issues/16.
This commit is contained in:
Lucas Verney 2016-03-10 18:02:56 +01:00
parent 3ec424081f
commit a736d41069
2 changed files with 1610 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1584
Examples.ipynb
View File

File diff suppressed because one or more lines are too long

31
replot/__init__.py
View File

@ -143,14 +143,14 @@ class Figure():
if len(args) == 0 and self.savepath is not None: if len(args) == 0 and self.savepath is not None:
args = (self.savepath,) args = (self.savepath,)
figure = self._render() figure = self.render()
figure.savefig(*args, **kwargs) figure.savefig(*args, **kwargs)
def show(self): def show(self):
""" """
Render and show the :class:`Figure` object. Render and show the :class:`Figure` object.
""" """
figure = self._render() figure = self.render()
figure.show() figure.show()
def set_grid(self, grid_description=None, def set_grid(self, grid_description=None,
@ -295,6 +295,13 @@ class Figure():
keyword will not affect the render unless you state \ keyword will not affect the render unless you state \
:mod:`replot` to use subplots. Note that ``_`` is a \ :mod:`replot` to use subplots. Note that ``_`` is a \
reserved group name which cannot be used. reserved group name which cannot be used.
- ``line`` which can be set to ``True``/``False`` to plot \
broken lines or discrete data series.
- ``logscale`` which can be either ``log`` or ``loglog`` to use \
such scales.
- ``orthonormal`` (boolean) to force axis to be orthonormal.
- ``xlim`` and ``ylim`` which are tuples of intervals on the \
x and y axis.
.. note:: Note that this API call considers list of tuples as \ .. note:: Note that this API call considers list of tuples as \
list of (x, y) coordinates to plot, contrary to standard \ list of (x, y) coordinates to plot, contrary to standard \
@ -524,7 +531,7 @@ class Figure():
else: else:
return cycler.cycler("color", self.palette) return cycler.cycler("color", self.palette)
def _render(self): def render(self):
""" """
Actually render the figure. Actually render the figure.
@ -556,6 +563,12 @@ class Figure():
elif plot_[2]["logscale"] == "loglog": elif plot_[2]["logscale"] == "loglog":
axis.set_xscale("log") axis.set_xscale("log")
axis.set_yscale("log") axis.set_yscale("log")
if "orthonormal" in plot_[2] and plot_[2]["orthonormal"]:
axis.set_aspect("equal")
if "xlim" in plot_[2]:
axis.set_xlim(*plot_[2]["xlim"])
if "ylim" in plot_[2]:
axis.set_ylim(*plot_[2]["ylim"])
# Do not clip line at the axes boundaries to prevent # Do not clip line at the axes boundaries to prevent
# extremas from being cropped. # extremas from being cropped.
for tmp_plot in tmp_plots: for tmp_plot in tmp_plots:
@ -747,6 +760,18 @@ def _handle_custom_plot_arguments(kwargs):
if "logscale" in kwargs: if "logscale" in kwargs:
custom_kwargs["logscale"] = kwargs["logscale"] custom_kwargs["logscale"] = kwargs["logscale"]
del kwargs["logscale"] del kwargs["logscale"]
# Handle "orthonormal" argument
if "orthonormal" in kwargs:
custom_kwargs["orthonormal"] = kwargs["orthonormal"]
del kwargs["orthonormal"]
# Handle "xlim" argument
if "xlim" in kwargs:
custom_kwargs["xlim"] = kwargs["xlim"]
del kwargs["xlim"]
# Handle "ylim" argument
if "ylim" in kwargs:
custom_kwargs["ylim"] = kwargs["ylim"]
del kwargs["ylim"]
return (kwargs, custom_kwargs) return (kwargs, custom_kwargs)