%pip install --upgrade ipympl speasy
try:
from google.colab import output
output.enable_custom_widget_manager()
except:
print("Not running inside Google Collab")
import speasy as spz
%matplotlib widget
amda_tree = spz.inventories.tree.amda
# Use this instead if you are not using jupyterlab yet
#%matplotlib notebook
import matplotlib.pyplot as plt
from datetime import datetime
plt.figure()
ace_mag = spz.get_data(amda_tree.Parameters.ACE.MFI.ace_imf_all.imf, datetime(2016,6,2), datetime(2016,6,5))
ace_mag.plot()
plt.tight_layout()
plt.show()