.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "examples/plot_version_diff_altitude.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        :ref:`Go to the end <sphx_glr_download_examples_plot_version_diff_altitude.py>`
        to download the full example code.

.. rst-class:: sphx-glr-example-title

.. _sphx_glr_examples_plot_version_diff_altitude.py:


Version Differences (Altitude)
------------------------------

This example demonstrates how to calculate the
percent change along an altitude profile between MSIS version 00 and 2.
Additionally, we show the difference between densities at noon (solid)
and midnight (dashed).

.. GENERATED FROM PYTHON SOURCE LINES 11-61



.. image-sg:: /examples/images/sphx_glr_plot_version_diff_altitude_001.png
   :alt: Longitude: 0, Latitude: 70
   :srcset: /examples/images/sphx_glr_plot_version_diff_altitude_001.png
   :class: sphx-glr-single-img





.. code-block:: Python


    import matplotlib.pyplot as plt
    import numpy as np

    import pymsis


    lon = 0
    lat = 70
    alts = np.linspace(0, 1000, 1000)
    f107 = 150
    f107a = 150
    ap = 7
    aps = [[ap] * 7]

    date = np.datetime64("2003-01-01T00:00")
    output_midnight2 = pymsis.calculate(date, lon, lat, alts, f107, f107a, aps)
    output_midnight0 = pymsis.calculate(date, lon, lat, alts, f107, f107a, aps, version=0)
    diff_midnight = (output_midnight2 - output_midnight0) / output_midnight0 * 100

    date = np.datetime64("2003-01-01T12:00")
    output_noon2 = pymsis.calculate(date, lon, lat, alts, f107, f107a, aps)
    output_noon0 = pymsis.calculate(date, lon, lat, alts, f107, f107a, aps, version=0)
    diff_noon = (output_noon2 - output_noon0) / output_noon0 * 100


    #  output is now of the shape (1, 1, 1, 1000, 11)
    # Get rid of the single dimensions
    diff_midnight = np.squeeze(diff_midnight)
    diff_noon = np.squeeze(diff_noon)

    _, ax = plt.subplots()
    for variable in pymsis.Variable:
        if variable.name in ("NO", "Total mass density", "Temperature"):
            # There is currently no NO data for earlier versions,
            # also ignore non-number densities
            continue
        (line,) = ax.plot(diff_midnight[:, variable], alts, linestyle="--")
        ax.plot(diff_noon[:, variable], alts, c=line.get_color(), label=variable.name)

    ax.legend(
        loc="upper center", bbox_to_anchor=(0.5, 1.15), fancybox=True, shadow=True, ncol=4
    )
    ax.set_title(f"Longitude: {lon}, Latitude: {lat}")
    ax.set_xlim(-50, 50)
    ax.set_ylim(0, 1000)
    ax.set_xlabel("Change from MSIS-00 to MSIS2 (%)")
    ax.set_ylabel("Altitude (km)")

    plt.show()


.. rst-class:: sphx-glr-timing

   **Total running time of the script:** (0 minutes 0.115 seconds)


.. _sphx_glr_download_examples_plot_version_diff_altitude.py:

.. only:: html

  .. container:: sphx-glr-footer sphx-glr-footer-example

    .. container:: sphx-glr-download sphx-glr-download-jupyter

      :download:`Download Jupyter notebook: plot_version_diff_altitude.ipynb <plot_version_diff_altitude.ipynb>`

    .. container:: sphx-glr-download sphx-glr-download-python

      :download:`Download Python source code: plot_version_diff_altitude.py <plot_version_diff_altitude.py>`

    .. container:: sphx-glr-download sphx-glr-download-zip

      :download:`Download zipped: plot_version_diff_altitude.zip <plot_version_diff_altitude.zip>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_