matplotlib/examples/units/units_scatter.py at pre-commit · ianhi/matplotlib · GitHub

Repository navigation

"""

=============

Unit handling

=============

The example below shows support for unit conversions over masked

arrays.

.. only:: builder_html

This example requires :download:`basic_units.py <basic_units.py>`

"""

import numpy as np

import matplotlib.pyplot as plt

from basic_units import secs, hertz, minutes

# create masked array

data = (1, 2, 3, 4, 5, 6, 7, 8)

mask = (1, 0, 1, 0, 0, 0, 1, 0)

xsecs = secs * np.ma.MaskedArray(data, mask, float)

fig, (ax1, ax2, ax3) = plt.subplots(nrows=3, sharex=True)

ax1.scatter(xsecs, xsecs)

ax1.yaxis.set_units(secs)

ax2.scatter(xsecs, xsecs, yunits=hertz)

ax3.scatter(xsecs, xsecs, yunits=minutes)

fig.tight_layout()

plt.show()