.. DO NOT EDIT. .. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY. .. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE: .. "examples/plot_surface_animation.py" .. LINE NUMBERS ARE GIVEN BELOW. .. only:: html .. note:: :class: sphx-glr-download-link-note :ref:`Go to the end ` to download the full example code. .. rst-class:: sphx-glr-example-title .. _sphx_glr_examples_plot_surface_animation.py: Surface animation ----------------- This example demonstrates how to animate the surface plot. Showing the variations over the course of a year. .. GENERATED FROM PYTHON SOURCE LINES 9-84 .. container:: sphx-glr-animation .. raw:: html

.. code-block:: Python import matplotlib import matplotlib.dates as mdates import matplotlib.pyplot as plt import numpy as np from matplotlib.animation import FuncAnimation import pymsis lons = range(-180, 185, 5) lats = range(-90, 95, 5) alt = 200 f107 = 150 f107a = 150 ap = 4 # Diurnal data dates = np.arange("2003-01-01", "2003-01-02", dtype="datetime64[30m]") ndates = len(dates) # (F107, F107a, ap) all need to be specified at the same length as dates f107s = [f107] * ndates f107as = [f107a] * ndates aps = [[ap] * 7] * ndates output = pymsis.calculate(dates, lons, lats, alt, f107s, f107as, aps) # output is now of the shape (ndates, nlons, nlats, 1, 11) # Get rid of the single dimensions output = np.squeeze(output) fig, (ax_time, ax_mesh) = plt.subplots( nrows=2, gridspec_kw={"height_ratios": [1, 4]}, constrained_layout=True ) xx, yy = np.meshgrid(lons, lats) vmin, vmax = 1e13, 8e13 norm = matplotlib.colors.Normalize(vmin, vmax) mesh = ax_mesh.pcolormesh( xx, yy, output[0, :, :, pymsis.Variable.N].T, shading="auto", norm=norm ) plt.colorbar( mesh, label=f"N number density at {alt} km (/m$^3$)", orientation="horizontal" ) time_data = output[:, len(lons) // 2, len(lats) // 2, :] ax_time.plot(dates, time_data[:, pymsis.Variable.N], c="k") ax_time.set_xlim(dates[0], dates[-1]) ax_time.set_ylim(vmin, vmax) ax_time.xaxis.set_major_locator(mdates.HourLocator(interval=3)) ax_time.xaxis.set_major_formatter(mdates.DateFormatter("%H:%M")) time_line = ax_time.axvline(dates[0], c="r") sun_loc = ( 180 - (dates.astype("datetime64[s]") - dates.astype("datetime64[D]")).astype(float) / 86400 * 360 ) (sun,) = ax_mesh.plot(sun_loc[0], 0, ".", c="gold", markersize=15) date_string = dates[0].astype("O").strftime("%H:%M") title = ax_time.set_title(f"{date_string}") ax_mesh.set_xlabel("Longitude") ax_mesh.set_ylabel("Latitude") def update_surface(t): date_string = dates[t].astype("O").strftime("%H:%M") title.set_text(f"{date_string}") time_line.set_xdata([dates[t]]) mesh.set_array(output[t, :, :, pymsis.Variable.N].T) sun.set_data([sun_loc[t]], [0]) ani = FuncAnimation(fig, update_surface, frames=range(ndates), interval=25) plt.show() .. rst-class:: sphx-glr-timing **Total running time of the script:** (0 minutes 39.925 seconds) .. _sphx_glr_download_examples_plot_surface_animation.py: .. only:: html .. container:: sphx-glr-footer sphx-glr-footer-example .. container:: sphx-glr-download sphx-glr-download-jupyter :download:`Download Jupyter notebook: plot_surface_animation.ipynb ` .. container:: sphx-glr-download sphx-glr-download-python :download:`Download Python source code: plot_surface_animation.py ` .. container:: sphx-glr-download sphx-glr-download-zip :download:`Download zipped: plot_surface_animation.zip ` .. only:: html .. rst-class:: sphx-glr-signature `Gallery generated by Sphinx-Gallery `_