Skip to content

Sliders to control matplotlib and other interactive goodies. Works in any interactive backend and even uses ipywidgets when in a Jupyter notebook

License

Notifications You must be signed in to change notification settings

jrussell25/mpl-interactions

 
 

Repository files navigation

mpl_interactions

All Contributors

Documentation StatusBinder (Warning: The interactions will be laggy when on binder)

Welcome!

mpl_interactions' library provides helpful ways to interact with Matplotlib plots. A summary of key components can be found below. Fuller narrative, further examples, and more information can be found on ReadtheDocs.

There are three submodules:

pyplot

Control Matplotlib plots using sliders and other widgets to adjust the parameters of the functions you are plotting. If working in a notebook then ipywidgets will be used to make the sliders, otherwise Matplotlib widgets will be used.

This is a different approach to controlling plots with sliders than ipywidgets.interact as when using interact you are responsible for:

  1. Defining the function to plot f(x,...) => y
  2. Handling the plotting logic (plt.plot, fig.cla, ax.set_ylim, etc)

In contrast, with mpl-interactions you only need to provide f(x, ...) => y and the plotting and updating boilerplate are handled for you.

x = np.linspace(0,6,100)
beta = np.linspace(0,5*np.pi)
def f(x, beta):
    return np.sin(x*4+beta)
interactive_plot(f, x=x, beta=beta)

These functions are designed to be used with ipympl, the backend that is designed for use in Jupyter Notebooks. So for optimal performance, make sure you set the backend with %matplotlib ipympl. That said, these functions will also work with any interactive backend (e.g. %matplotlib qt5).

generic

Provides ways to interact with Matplotlib that will work outside of a Jupyter Notebook; this should work equally well with any backend.

  1. A very niche (but very cool) way to compare 2D heatmaps
  2. Scroll to zoom
  3. Middle click to pan

utils

This module includes utility functions to make things just that little bit easier.

  1. ioff as a context manager
from mpl_interactions.utils import ioff
with ioff:
    # interactive mode will be off
    fig = plt.figure()
    # other stuff
# interactive mode will be on
  1. figure that accepts a scalar for figsize (this will scale the default dimensions)
from mpl_interactions.utils import figure
fig = figure(3)
# the default figsize is [6.4, 4.8], this figure will have figsize = [6.4*3, 4.8*3]
  1. nearest_idx -- avoid ever having to write np.argmin(np.abs(arr - value)) again

Installation

pip install mpl_interactions

# if using jupyterlab
conda install nodejs=12
jupyter labextension install @jupyter-widgets/jupyterlab-manager

If you use JupyterLab, make sure you follow the full instructions in the ipympl readme in particular installing jupyterlab-manager.

Contributing / feature requests / roadmap

I use the GitHub issues to keep track of ideas I have, so looking through those should serve as a roadmap of sorts. For the most part I add to the library when I create a function that is useful for the science I am doing. If you create something that seems useful a PR would be most welcome so we can share it easily with more people. I'm also open to feature requests if you have an idea.

Documentation

The fuller narrative documentation can be found on ReadTheDocs. You may also find it helpful to check out the examples directory.

Examples with GIFs!

Tragically, neither GitHub nor the sphinx documentation render the actual moving plots so instead, here are gifs of the functions. The code for these can be found in the notebooks in the examples directory.

interactive_plot

Easily make a line plot interactive:

heatmap_slicer

Compare vertical and horizontal slices across multiple heatmaps:

scrolling zoom + middle click pan

Contributors ✨

Thanks goes to these wonderful people (emoji key):


Ian Hunt-Isaak

💻

Sam

📖

Jenny Coulter

📓

Sabina Haque

📖 📓 💻

This project follows the all-contributors specification. Contributions of any kind welcome!

About

Sliders to control matplotlib and other interactive goodies. Works in any interactive backend and even uses ipywidgets when in a Jupyter notebook

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.8%
  • Shell 0.2%