1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117
|
Plotting Astropy objects in Matplotlib
**************************************
.. _plotting-quantities:
Plotting quantities
===================
|Quantity| objects can be conveniently plotted using matplotlib. This
feature needs to be explicitly turned on:
.. doctest-requires:: matplotlib
>>> from astropy.visualization import quantity_support
>>> quantity_support() # doctest: +IGNORE_OUTPUT
<astropy.visualization.units.MplQuantityConverter ...>
Then |Quantity| objects can be passed to matplotlib plotting
functions. The axis labels are automatically labeled with the unit of
the quantity:
.. plot::
:include-source:
:context: reset
from astropy import units as u
from astropy.visualization import quantity_support
quantity_support()
from matplotlib import pyplot as plt
plt.figure(figsize=(5,3))
plt.plot([1, 2, 3] * u.m)
Quantities are automatically converted to the first unit set on a
particular axis, so in the following, the y-axis remains in ``m`` even
though the second line is given in ``cm``:
.. plot::
:include-source:
:context:
plt.plot([1, 2, 3] * u.cm)
Plotting a quantity with an incompatible unit will raise an exception.
For example, calling ``plt.plot([1, 2, 3] * u.kg)`` (mass unit) to overplot
on the plot above that is displaying length units.
To make sure unit support is turned off afterward, you can use
`~astropy.visualization.quantity_support` with a ``with`` statement::
with quantity_support():
plt.plot([1, 2, 3] * u.m)
.. _plotting-times:
Plotting times
==============
Matplotlib natively provides a mechanism for plotting dates and times on one
or both of the axes, as described in
`Date tick labels <https://matplotlib.org/stable/gallery/text_labels_and_annotations/date.html>`_.
To make use of this, you can use the ``plot_date`` attribute of |Time| to get
values in the time system used by Matplotlib.
However, in many cases, you will probably want to have more control over the
precise scale and format to use for the tick labels, in which case you can make
use of the `~astropy.visualization.time_support` function. This feature needs to
be explicitly turned on:
.. doctest-requires:: matplotlib
>>> from astropy.visualization import time_support
>>> time_support() # doctest: +IGNORE_OUTPUT
<astropy.visualization.units.MplTimeConverter ...>
Once this is enabled, |Time| objects can be passed to matplotlib plotting
functions. The axis labels are then automatically labeled with times formatted
using the |Time| class:
.. plot::
:include-source:
:context: reset
from matplotlib import pyplot as plt
from astropy.time import Time
from astropy.visualization import time_support
time_support()
plt.figure(figsize=(5,3))
plt.plot(Time([58000, 59000, 62000], format='mjd'), [1.2, 3.3, 2.3])
By default, the format and scale used for the plots is taken from the first time
that Matplotlib encounters for a particular Axes instance. The format and scale
can also be explicitly controlled by passing arguments to ``time_support``:
.. plot::
:nofigs:
:context: reset
from matplotlib import pyplot as plt
from astropy.time import Time
from astropy.visualization import time_support
.. plot::
:include-source:
:context:
time_support(format='mjd', scale='tai')
plt.figure(figsize=(5,3))
plt.plot(Time([50000, 52000, 54000], format='mjd'), [1.2, 3.3, 2.3])
To make sure support for plotting times is turned off afterward, you can use
`~astropy.visualization.time_support` as a context manager::
with time_support(format='mjd', scale='tai'):
plt.figure(figsize=(5,3))
plt.plot(Time([50000, 52000, 54000], format='mjd'))
|