In [1]:
%matplotlib notebook
In [2]:
import os
import sys
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.ticker import (MultipleLocator, FormatStrFormatter, AutoMinorLocator)
import pandas as pd
In [3]:
os.chdir('..')
os.getcwd()
Out[3]:
In [4]:
sys.path.append('../scripts/')
import bicorr_plot as bicorr_plot
import bicorr_math as bicorr_math
In [5]:
os.listdir('cgmf/datap/')
Out[5]:
In [6]:
Esum_df_meas = pd.read_csv(r'Cf072115_to_Cf072215b/datap/Esum_df.csv',index_col=0)
Esum_df_cgmf = pd.read_csv(r'cgmf/datap/Esum_df.csv',index_col=0)
Esum_df_freya= pd.read_csv(r'freya/datap/Esum_df.csv',index_col=0)
Esum_df_ipol = pd.read_csv(r'ipol/datap/Esum_df.csv',index_col=0)
Esum_df_ipol_noct = pd.read_csv(r'ipol_noct/datap/Esum_df.csv',index_col=0)
In [7]:
Esum_df_ipol
Out[7]:
In [8]:
Esum_dfs = [Esum_df_meas, Esum_df_cgmf, Esum_df_freya, Esum_df_ipol, Esum_df_ipol_noct]
legends =['Experiment', 'CGMF', 'FREYA', 'PoliMi', 'PoliMi-No CT']
fmts = ['x', 's', 'D', 'o', '^']
colors = ['#5d269b', '#dd673b', '#80bc31', '#3cbfe0', '#4242f4']
to_plot = [0,1, 2, 3]
In [9]:
line_thickness = 1
ebar_width = 3
In [13]:
fig = plt.figure(figsize=(4,4))
ax = plt.gca()
for i in to_plot:
Esum_df = Esum_dfs[i]
ax.errorbar(Esum_df['th_bin_center'],
Esum_df['Eave'],
yerr=Esum_df['Eave_err'],
fmt=fmts[i],
markeredgewidth=1,
markerfacecolor='none',
elinewidth=line_thickness,
capthick = line_thickness,
capsize = ebar_width,
c=colors[i])
leg = plt.legend([legends[i] for i in to_plot])
leg.get_frame().set_edgecolor('w')
ax.axvspan(0,20,facecolor='gray', alpha=0.2)
ax.set_xlabel('Angle (degrees)')
ax.set_ylabel('$\overline{E_n}$ (MeV)')
ax.set_xlim([0,180])
# ax.set_ylim([2.5,3.3])
# Set up ticks
ax.tick_params(axis='both',
which='major',
direction='inout',
length=6,
color='k',
bottom=True, right=True, top=True, left=True)
ax.tick_params(axis='both',
which='minor',
direction='in',
length=3,
bottom=True, right=True, top=True, left=True)
# Major
ax.xaxis.set_major_locator(MultipleLocator(45))
ax.yaxis.set_major_locator(MultipleLocator(.05))
# Minor
ax.xaxis.set_minor_locator(MultipleLocator(15))
ax.yaxis.set_minor_locator(MultipleLocator(0.025))
ax.text(30,2.42,'(a)', size=15, backgroundcolor='white')
plt.tight_layout()
bicorr_plot.save_fig_to_folder('Esum_vs_angle_compare',r'compare/fig')
In [14]:
Esum_exp = Esum_dfs[0]
to_plot = [1,2,3]
fig = plt.figure(figsize=(4,4))
ax = plt.gca()
for i in to_plot:
Esum_df = Esum_dfs[i]
x = Esum_df['th_bin_center']
# y = Esum_df['Eave']/Esum_exp['Eave']
y, yerr = bicorr_math.prop_err_division(Esum_df['Eave'],Esum_df['Eave_err'],
Esum_exp['Eave'],Esum_exp['Eave_err'])
ax.errorbar(x,
y,
yerr=yerr,
fmt=fmts[i],
markeredgewidth=1,
markerfacecolor='none',
elinewidth=line_thickness,
capthick = line_thickness,
capsize = ebar_width,
c=colors[i])
leg = plt.legend([legends[i] for i in to_plot],loc=9)
leg.get_frame().set_edgecolor('w')
plt.axhline(1.0,color='gray', linewidth=1,linestyle='--')
ax.axvspan(0,20,facecolor='gray', alpha=0.2)
ax.set_xlabel('Angle (degrees)')
ax.set_ylabel(r'$\left[\overline{E_n}\right]_{SIM} / \left[\overline{E_n}\right]_{EXP}$')
#plt.ylabel(r'$\Big[ \overline{E_n} \Big]_{\texttt{EXP}}$')
ax.set_xlim([0,180])
# ax.set_ylim([2.5,3.3])
# Set up ticks
ax.tick_params(axis='both',
which='major',
direction='inout',
length=6,
color='k',
bottom=True, right=True, top=True, left=True)
ax.tick_params(axis='both',
which='minor',
direction='in',
length=3,
bottom=True, right=True, top=True, left=True)
# Major
ax.xaxis.set_major_locator(MultipleLocator(45))
ax.yaxis.set_major_locator(MultipleLocator(.02))
# Minor
ax.xaxis.set_minor_locator(MultipleLocator(15))
ax.yaxis.set_minor_locator(MultipleLocator(0.005))
ax.text(30,1.02,'(b)', size=15, backgroundcolor='white')
plt.tight_layout()
bicorr_plot.save_fig_to_folder('Esum_vs_angle_diff',r'compare/fig')
plt.show()
In [ ]: