# -*- coding: utf-8 -*- """ Created on Thu Jul 16 12:50:21 2020 @author: baum_c4 """ import numpy as np import pandas as pd import pickle import matplotlib.colors as mcol import matplotlib.cm as cm import matplotlib.pyplot as plt import matplotlib with open('plot_learningset.pickle', 'rb') as f: [learnvector_o,learnvector_m,learnvector_s,timevector]=pickle.load(f) posx_o=learnvector_o[0] posy_o=learnvector_o[1] posz_o=learnvector_o[2] velx_o=learnvector_o[3] vely_o=learnvector_o[4] velz_o=learnvector_o[5] dst=learnvector_o[6] cm1 = mcol.LinearSegmentedColormap.from_list("MyCmapName",["b","m","r"]) f,ax = plt.subplots(1,1,figsize=(14*0.39,8*0.39)) c=ax.scatter(posx_o/6372,posy_o/6371,c=posz_o/6371,vmin=-22,vmax=22,cmap=cm1) cbar=plt.colorbar(c) cbar.set_label('ACE position in Z [Re]') #ax.set_rmax(2) #ax.set_rticks([0.5, 1, 1.5, 2]) # less radial ticks #ax.set_rlabel_position(-22.5) # get radial labels away from plotted line ax.set_xlabel('ACE position in X [Re]') ax.set_ylabel('ACE position in Z [Re]') ax.set_title('ACE Position in GSE') plt.xticks(np.arange(215, 256, step=10)) plt.yticks(np.arange(-45, 46, step=15)) ax.grid(True) plt.tight_layout() plt.savefig('pos_learn.pdf',bbox_inches='tight') plt.show() font = { 'size' : 12} matplotlib.rc('font', **font) cm1 = mcol.LinearSegmentedColormap.from_list("MyCmapName",["b","m","r"]) f,ax = plt.subplots(2,1,figsize=(16*0.39,15*0.39)) c=ax[0].scatter(posx_o/6372,posy_o/6371,c=posz_o/6371,vmin=-22,vmax=22,cmap=cm1) cbar=f.colorbar(c,ax=ax[0]) cbar.set_label('ACE position in Z [Re]') #ax.set_rmax(2) #ax.set_rticks([0.5, 1, 1.5, 2]) # less radial ticks #ax.set_rlabel_position(-22.5) # get radial labels away from plotted line ax[0].set_xlabel('ACE position in X [Re]') ax[0].set_ylabel('ACE position in Z [Re]') ax[0].set_title('ACE Position in GSE') plt.xticks(np.arange(215, 256, step=10)) plt.yticks(np.arange(-45, 46, step=15)) ax[0].grid(True) c=ax[1].scatter(velx_o,vely_o,c=timevector[:,2]/60,cmap=cm1,vmin=10,vmax=90) cbar=plt.colorbar(c,ax=ax[1],ticks=[10,30,50,70,90]) cbar.set_label('meas. propagation delay [min]') ax[1].set_title('Solar wind speed in GSE') plt.xticks(np.arange(-1000, -199, step=200)) plt.yticks(np.arange(-150, 151, step=50)) #ax.set_rmax(2) #ax.set_rticks([0.5, 1, 1.5, 2]) # less radial ticks #ax.set_rlabel_position(-22.5) # get radial labels away from plotted line ax[1].set_xlabel('Solar wind speed in X [km/s]') ax[1].set_ylabel('Solar wind speed in Y [km/s]') ax[1].grid(True) plt.tight_layout() plt.savefig('plot_learn.pdf',bbox_inches='tight') plt.show()