Skip to content
Snippets Groups Projects
Commit 322b5d6d authored by Guillaume Demesy's avatar Guillaume Demesy
Browse files

tweaks M matrix plots

parent 4811328d
Branches
No related tags found
No related merge requests found
Pipeline #10108 passed
import numpy as np import numpy as np
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import matplotlib as mpl import matplotlib as mpl
mpl.font_manager._rebuild() plt.rcParams.update({"text.usetex": True, "font.family": "serif"})
plt.rcParams.update({"text.usetex": True})
def add_colorbar(mappable): def add_colorbar(mappable):
from mpl_toolkits.axes_grid1 import make_axes_locatable from mpl_toolkits.axes_grid1 import make_axes_locatable
...@@ -43,7 +42,7 @@ effr2_sin = np.zeros((9,nb_lam,nb_phi),dtype=complex) ...@@ -43,7 +42,7 @@ effr2_sin = np.zeros((9,nb_lam,nb_phi),dtype=complex)
Qscat_sin = np.zeros((nb_lam,nb_phi)) Qscat_sin = np.zeros((nb_lam,nb_phi))
tab_lam = np.linspace(400,800,nb_lam) tab_lam = np.linspace(350,800,nb_lam)
tab_phi = np.linspace(0,360,nb_phi) tab_phi = np.linspace(0,360,nb_phi)
M = np.zeros((4,4,len(tab_lam),len(tab_phi)),dtype=complex) M = np.zeros((4,4,len(tab_lam),len(tab_phi)),dtype=complex)
...@@ -146,18 +145,24 @@ for i in range(4): ...@@ -146,18 +145,24 @@ for i in range(4):
ax.set_yticks([]) ax.set_yticks([])
cbar = plt.colorbar(sm, ax=ax, fraction=0.046, pad=0.04) cbar = plt.colorbar(sm, ax=ax, fraction=0.046, pad=0.04)
cbar.ax.locator_params(nbins=5) cbar.ax.locator_params(nbins=5)
cbar.ax.tick_params(labelsize=14)
else: else:
p00 = ax.contourf(tab_phi*np.pi/180,which_r,M[i,j]/M[0,0]-1,cmap=plt.cm.bwr,vmin=-1,vmax=1) p00 = ax.contourf(tab_phi*np.pi/180,which_r,M[i,j]/M[0,0]-1,cmap=plt.cm.bwr,vmin=-1,vmax=1)
label_position=ax.get_rlabel_position() ax.xaxis.label.set_color('C0') #setting up X-axis label color to yellow
ax.text(np.radians(label_position+30),ax.get_rmax()/1.3,rlabel, ax.yaxis.label.set_color('C3') #setting up Y-axis label color to blue
rotation=label_position,ha='center',va='center',fontsize=16) ax.tick_params(axis='x', colors='C0') #setting up X-axis tick color to red
ax.text(np.radians(-15),tab_lam.max()*1.01,r"$\varphi_0$",fontsize=16) ax.tick_params(axis='y', colors='C3') #setting up Y-axis tick color to black
ax.set_rlabel_position(70)
ax.tick_params(axis='both', which='major', labelsize=14)
ax.text(np.radians(22.5),tab_lam.max()*1.03,r"$\varphi_0$",fontsize=16,color='C0')
ax.text(np.radians(90),ax.get_rmax()/1.3,rlabel,
rotation=70,ha='center',va='center',fontsize=16,color='C3')
norm = mpl.colors.Normalize(vmin=-1,vmax=1) norm = mpl.colors.Normalize(vmin=-1,vmax=1)
sm = mpl.cm.ScalarMappable(cmap=plt.cm.bwr, norm=norm) sm = mpl.cm.ScalarMappable(cmap=plt.cm.bwr, norm=norm)
sm.set_array([]) sm.set_array([])
ax.set_rorigin(which_orig) ax.set_rorigin(which_orig)
plt.subplots_adjust(top=0.92, bottom=0.08, left=0.10, right=0.95, hspace=0.25,wspace=0.35) plt.subplots_adjust(top=0.92, bottom=0.08, left=0.10, right=0.95, hspace=0.25,wspace=0.35)
plt.savefig('Mmatrix.jpg') # plt.savefig('Mmatrix.jpg')
plt.savefig('Mmatrix.svg') plt.savefig('Mmatrix.pdf',bbox_inches='tight',pad_inches=0)
plt.show() plt.show()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment