nucleardatapy 0.2.0__py3-none-any.whl → 1.0.0__py3-none-any.whl

nucleardatapy/fig/matter_setupMicro_LP_fig.py

@@ -3,7 +3,8 @@ import matplotlib.pyplot as plt
 
 import nucleardatapy as nuda
 
-def matter_setupMicro_LP_fig( pname, models, matter = 'SM', ell = 0 ):
+
+def matter_setupMicro_LP_fig( pname, models, matter="SM", ell=0 ):
     """
     Plot nucleonic energy per particle E/A in matter.\
     The plot is 2x2 with:\
@@ -21,89 +22,185 @@ def matter_setupMicro_LP_fig( pname, models, matter = 'SM', ell = 0 ):
 
     """
     #
-    print(f'Plot name: {pname}')
+    print(f"Plot name: {pname}")
     #
-    if matter.lower() == 'sm':
-    	fig, axs = plt.subplots(2,2)
-    	fig.subplots_adjust(left=0.12, bottom=0.12, right=0.98, top=0.98, wspace=0.15, hspace=0.05 )
-    elif matter.lower() == 'nm':
-    	fig, axs = plt.subplots(1,2)
-    	fig.subplots_adjust(left=0.12, bottom=0.12, right=0.98, top=0.98, wspace=0.15, hspace=0.3 )
+    if matter.lower() == "sm":
+        fig, axs = plt.subplots(2, 2)
+        fig.subplots_adjust( left=0.12, bottom=0.12, right=0.98, top=0.98, wspace=0.15, hspace=0.05 )
+    elif matter.lower() == "nm":
+        fig, axs = plt.subplots(1, 2)
+        fig.subplots_adjust( left=0.12, bottom=0.12, right=0.98, top=0.98, wspace=0.15, hspace=0.05 )
     else:
-    	print('matter_setupMicro_LP_fig: issue with matter:',matter)
-    	print('matter_setupMicro_LP_fig: --- exit() ---')
-    	exit()
+        print("matter_setupMicro_LP_fig: issue with matter:", matter)
+        print("matter_setupMicro_LP_fig: --- exit() ---")
+        exit()
     #
     #
-    if matter.lower() == 'sm':
-    	axs[0,0].set_xlim([0, 2.0])
-    	axs[0,0].set_ylim([-2.1, 1.6])
-    	if ell == 0: axs[0,0].set_ylabel(r'$F_0$')
-    	if ell == 1: axs[0,0].set_ylabel(r'$F_1$')
-    	axs[0,0].tick_params('x', labelbottom=False)
-    	if ell == 0: axs[0,1].set_ylabel(r'$G_0$')
-    	if ell == 1: axs[0,1].set_ylabel(r'$G_1$')
-    	axs[0,1].set_xlim([0, 2.0])
-    	axs[0,1].set_ylim([-2.1, 1.6])
-    	axs[0,1].tick_params('x', labelbottom=False)
-    	axs[0,1].tick_params('y', labelleft=False)
-    	if ell == 0: axs[1,0].set_ylabel(r'$F_0^\prime$')
-    	if ell == 1: axs[1,0].set_ylabel(r'$F_1^\prime$')
-    	axs[1,0].set_xlabel(r'$k_F$ (fm$^{-1}$)')
-    	axs[1,0].set_xlim([0, 2.0])
-    	axs[1,0].set_ylim([-2.1, 1.6])
-    	if ell == 0: axs[1,1].set_ylabel(r'$G_0^\prime$')
-    	if ell == 1: axs[1,1].set_ylabel(r'$G_1^\prime$')
-    	axs[1,1].set_xlabel(r'$k_F$ (fm$^{-1}$)')
-    	axs[1,1].set_xlim([0, 2.0])
-    	axs[1,1].set_ylim([-2.1, 1.6])
-    	axs[1,1].tick_params('y', labelleft=False)
-    elif matter.lower() == 'nm':
-    	if ell == 0: axs[0].set_ylabel(r'$F_0$')
-    	if ell == 1: axs[0].set_ylabel(r'$F_1$')
-    	axs[0].set_xlabel(r'$k_{F_n}$ (fm$^{-1}$)')
-    	axs[0].set_xlim([0, 2.0])
-    	axs[0].set_ylim([-1.1, 1.1])
-    	if ell == 0: axs[1].set_ylabel(r'$G_0$')
-    	if ell == 1: axs[1].set_ylabel(r'$G_1$')
-    	axs[1].set_xlabel(r'$k_{F_n}$ (fm$^{-1}$)')
-    	axs[1].set_xlim([0, 2.0])
-    	axs[1].set_ylim([-1.1, 1.1])
-    	axs[1].tick_params('y', labelleft=False)
+    if matter.lower() == "sm":
+        axs[0, 0].set_xlim([0, 2.0])
+        axs[0, 0].set_ylim([-2.1, 1.6])
+        if ell == 0:
+            axs[0, 0].set_ylabel(r"$F_0$", fontsize='14')
+        if ell == 1:
+            axs[0, 0].set_ylabel(r"$F_1$", fontsize='14')
+        axs[0, 0].tick_params("x", labelbottom=False)
+        if ell == 0:
+            axs[0, 1].set_ylabel(r"$G_0$", fontsize='14')
+        if ell == 1:
+            axs[0, 1].set_ylabel(r"$G_1$", fontsize='14')
+        axs[0, 1].set_xlim([0, 2.0])
+        axs[0, 1].set_ylim([-2.1, 1.6])
+        axs[0, 1].tick_params("x", labelbottom=False)
+        axs[0, 1].tick_params("y", labelleft=False)
+        if ell == 0:
+            axs[1, 0].set_ylabel(r"$F_0^\prime$", fontsize='14')
+        if ell == 1:
+            axs[1, 0].set_ylabel(r"$F_1^\prime$", fontsize='14')
+        axs[1, 0].set_xlabel(r"$k_{F}$ (fm$^{-1}$)", fontsize='14')
+        axs[1, 0].set_xlim([0, 2.0])
+        axs[1, 0].set_ylim([-1.1, 2.6])
+        if ell == 0:
+            axs[1, 1].set_ylabel(r"$G_0^\prime$", fontsize='14')
+        if ell == 1:
+            axs[1, 1].set_ylabel(r"$G_1^\prime$", fontsize='14')
+        axs[1, 1].set_xlabel(r"$k_{F}$ (fm$^{-1}$)", fontsize='14')
+        axs[1, 1].set_xlim([0, 2.0])
+        axs[1, 1].set_ylim([-1.1, 2.6])
+        axs[1, 1].tick_params("y", labelleft=False)
+    elif matter.lower() == "nm":
+        if ell == 0:
+            axs[0].set_ylabel(r"$F_0$", fontsize='14')
+        if ell == 1:
+            axs[0].set_ylabel(r"$F_1$", fontsize='14')
+        axs[0].set_xlabel(r"$k_{F}$ (fm$^{-1}$)", fontsize='14')
+        axs[0].set_xlim([0, 2.0])
+        axs[0].set_ylim([-1.1, 1.1])
+        if ell == 0:
+            axs[1].set_ylabel(r"$G_0$", fontsize='14')
+        if ell == 1:
+            axs[1].set_ylabel(r"$G_1$", fontsize='14')
+        axs[1].set_xlabel(r"$k_{F}$ (fm$^{-1}$)", fontsize='14')
+        axs[1].set_xlim([0, 2.0])
+        axs[1].set_ylim([-1.1, 1.1])
+        axs[1].tick_params("y", labelleft=False)
     #
     for model in models:
+        #
+        mic = nuda.matter.setupMicroLP(model=model)
+        #
+        if matter in model:
+            print("\nmodel:", model, "\n")
+            if mic.err:
+                if matter.lower() == "sm" and mic.sm_LP["F"][ell] is not None:
+                    axs[0, 0].errorbar(
+                        mic.sm_kfn,
+                        mic.sm_LP["F"][ell],
+                        yerr=mic.sm_LP_F_err[ell],
+                        marker=mic.marker,
+                        linestyle="none",
+                        label=mic.label,
+                    )
+                    axs[0, 1].errorbar(
+                        mic.sm_kfn,
+                        mic.sm_LP["G"][ell],
+                        yerr=mic.sm_LP_G_err[ell],
+                        marker=mic.marker,
+                        linestyle="none",
+                        label=mic.label,
+                    )
+                    axs[1, 0].errorbar(
+                        mic.sm_kfn,
+                        mic.sm_LP["Fp"][ell],
+                        yerr=mic.sm_LP_Fp_err[ell],
+                        marker=mic.marker,
+                        linestyle="none",
+                        label=mic.label,
+                    )
+                    axs[1, 1].errorbar(
+                        mic.sm_kfn,
+                        mic.sm_LP["Gp"][ell],
+                        yerr=mic.sm_LP_Gp_err[ell],
+                        marker=mic.marker,
+                        linestyle="none",
+                        label=mic.label,
+                    )
+                elif matter.lower() == "nm" and mic.nm_LP["F"][ell] is not None:
+                    axs[0].errorbar(
+                        mic.nm_kfn,
+                        mic.nm_LP["F"][ell],
+                        marker=mic.marker,
+                        linestyle="none",
+                        label=mic.label,
+                    )
+                    axs[1].errorbar(
+                        mic.nm_kfn,
+                        mic.nm_LP["G"][ell],
+                        marker=mic.marker,
+                        linestyle="none",
+                        label=mic.label,
+                    )
+            else:
+                if matter.lower() == "sm" and mic.sm_LP["F"][ell] is not None:
+                    axs[0, 0].plot(
+                        mic.sm_kfn,
+                        mic.sm_LP["F"][ell],
+                        marker=mic.marker,
+                        linestyle=mic.linestyle,
+                        markevery=mic.every,
+                        label=mic.label,
+                    )
+                    axs[0, 1].plot(
+                        mic.sm_kfn,
+                        mic.sm_LP["G"][ell],
+                        marker=mic.marker,
+                        linestyle=mic.linestyle,
+                        markevery=mic.every,
+                        label=mic.label,
+                    )
+                    axs[1, 0].plot(
+                        mic.sm_kfn,
+                        mic.sm_LP["Fp"][ell],
+                        marker=mic.marker,
+                        linestyle=mic.linestyle,
+                        markevery=mic.every,
+                        label=mic.label,
+                    )
+                    axs[1, 1].plot(
+                        mic.sm_kfn,
+                        mic.sm_LP["Gp"][ell],
+                        marker=mic.marker,
+                        linestyle=mic.linestyle,
+                        markevery=mic.every,
+                        label=mic.label,
+                    )
+                elif matter.lower() == "nm" and mic.nm_LP["F"][ell] is not None:
+                    axs[0].plot(
+                        mic.nm_kfn,
+                        mic.nm_LP["F"][ell],
+                        marker=mic.marker,
+                        linestyle=mic.linestyle,
+                        markevery=mic.every,
+                        label=mic.label,
+                    )
+                    axs[1].plot(
+                        mic.nm_kfn,
+                        mic.nm_LP["G"][ell],
+                        marker=mic.marker,
+                        linestyle=mic.linestyle,
+                        markevery=mic.every,
+                        label=mic.label,
+                    )
+        #
+        if nuda.env.verb_output:
+            mic.print_outputs()
+        #
     #
-    	mic = nuda.matter.setupMicroLP( model = model )
-    	if matter in model:
-    		print('\nmodel:',model,'\n')
-    		if mic.err:
-    			if matter.lower() == 'sm' and mic.sm_kfn is not None:
-    				axs[0,0].errorbar( mic.sm_kfn, mic.sm_LP['F'][ell], yerr=mic.sm_LP_F_err[ell], marker=mic.marker, linestyle='none', label=mic.label )
-    				axs[0,1].errorbar( mic.sm_kfn, mic.sm_LP['G'][ell], yerr=mic.sm_LP_G_err[ell], marker=mic.marker, linestyle='none', label=mic.label )
-    				axs[1,0].errorbar( mic.sm_kfn, mic.sm_LP['Fp'][ell], yerr=mic.sm_LP_Fp_err[ell], marker=mic.marker, linestyle='none', label=mic.label )
-    				axs[1,1].errorbar( mic.sm_kfn, mic.sm_LP['Gp'][ell], yerr=mic.sm_LP_Gp_err[ell], marker=mic.marker, linestyle='none', label=mic.label )
-    			elif matter.lower() == 'nm' and mic.nm_kfn is not None:
-    				axs[0].errorbar( mic.nm_kfn, mic.nm_LP['F'][ell], marker=mic.marker, linestyle='none', label=mic.label )
-    				axs[1].errorbar( mic.nm_kfn, mic.nm_LP['G'][ell], marker=mic.marker, linestyle='none', label=mic.label )
-    		else:
-    			if matter.lower() == 'sm' and mic.sm_kfn is not None:
-    				axs[0,0].plot( mic.sm_kfn, mic.sm_LP['F'][ell], marker=mic.marker, linestyle=mic.linestyle, markevery=mic.every, label=mic.label )
-    				axs[0,1].plot( mic.sm_kfn, mic.sm_LP['G'][ell], marker=mic.marker, linestyle=mic.linestyle, markevery=mic.every, label=mic.label )
-    				axs[1,0].plot( mic.sm_kfn, mic.sm_LP['Fp'][ell], marker=mic.marker, linestyle=mic.linestyle, markevery=mic.every, label=mic.label )
-    				axs[1,1].plot( mic.sm_kfn, mic.sm_LP['Gp'][ell], marker=mic.marker, linestyle=mic.linestyle, markevery=mic.every, label=mic.label )
-    			elif matter.lower() == 'nm' and mic.nm_kfn is not None:
-    				axs[0].plot( mic.nm_kfn, mic.nm_LP['F'][ell], marker=mic.marker, linestyle=mic.linestyle, markevery=mic.every, label=mic.label )
-    				axs[1].plot( mic.nm_kfn, mic.nm_LP['G'][ell], marker=mic.marker, linestyle=mic.linestyle, markevery=mic.every, label=mic.label )
-    	#
-    	if nuda.env.verb_output: mic.print_outputs( )
-    	#
-    #
-    if matter.lower() == 'sm':
-    	axs[1,0].legend(loc='lower left',fontsize='6', ncol=2)
-    elif matter.lower() == 'nm':
-    	axs[0].legend(loc='upper left',fontsize='8', ncol=1)
+    if matter.lower() == "sm":
+        axs[1, 0].legend(loc="upper left", fontsize="6", ncol=2)
+    elif matter.lower() == "nm":
+        axs[0].legend(loc="upper left", fontsize="8", ncol=1)
     #
     if pname is not None:
-    	plt.savefig(pname, dpi=300)
-    	plt.close()
-    #
+        plt.savefig(pname, dpi=300)
+        plt.close()
+    #

nucleardatapy/fig/matter_setupMicro_band_fig.py

@@ -3,7 +3,8 @@ import matplotlib.pyplot as plt
 
 import nucleardatapy as nuda
 
-def matter_setupMicro_band_fig( pname, models, den, matter ):
+
+def matter_setupMicro_band_fig(pname, models, den, matter):
     """
     Plot the correlation between Esym and Lsym.\
     The plot is 1x1 with:\
@@ -16,70 +17,138 @@ def matter_setupMicro_band_fig( pname, models, den, matter ):
 
     """
     #
-    print(f'Plot name: {pname}')
+    print(f"Plot name: {pname}")
     #
-    #den = 0.03*np.arange(13)
+    # den = 0.03*np.arange(13)
     #
-    fig, axs = plt.subplots(1,2)
-    fig.tight_layout() # Or equivalently,  "plt.tight_layout()"
-    fig.subplots_adjust(left=0.15, bottom=0.12, right=None, top=0.98, wspace=0.38, hspace=0.3 )
+    fig, axs = plt.subplots(1, 2)
+    fig.tight_layout()  # Or equivalently,  "plt.tight_layout()"
+    fig.subplots_adjust( left=0.15, bottom=0.12, right=0.95, top=0.98, wspace=0.38, hspace=0.3 )
     #
-    axs[0].set_xlabel(r'$n_\text{nuc}$ (fm$^{-3}$)',fontsize='14')
+    axs[0].set_xlabel(r"$n_\text{nuc}$ (fm$^{-3}$)", fontsize="14")
     axs[0].set_xlim([0, 0.2])
     #
-    axs[1].set_xlabel(r'$n_\text{nuc}$ (fm$^{-3}$)',fontsize='14')
+    axs[1].set_xlabel(r"$n_\text{nuc}$ (fm$^{-3}$)", fontsize="14")
     axs[1].set_xlim([0, 0.2])
     #
-    if matter.lower() == 'nm':
-        axs[0].set_ylabel(r'$E_\text{NM}$',fontsize='14')
-        axs[1].set_ylabel(r'$E_\text{NM}/E_\text{NM,NRFFG}$',fontsize='14')
+    if matter.lower() == "nm":
+        axs[0].set_ylabel(r"$E_\text{int,NM}$", fontsize="14")
+        axs[1].set_ylabel(r"$E_\text{int,NM}/E_\text{int,NM}^\text{NR FFG}$", fontsize="14")
         axs[0].set_ylim([0, 20])
         axs[1].set_ylim([0.2, 0.9])
-    elif matter.lower() == 'sm':
-        axs[0].set_ylabel(r'$E_\text{SM}$',fontsize='14')
-        axs[1].set_ylabel(r'$E_\text{SM}/E_\text{SM,NRFFG}$',fontsize='14')
+    elif matter.lower() == "sm":
+        axs[0].set_ylabel(r"$E_\text{int,SM}$", fontsize="14")
+        axs[1].set_ylabel(r"$E_\text{int,SM}/E_\text{int,SM}^\text{NR FFG}$", fontsize="14")
         axs[0].set_ylim([-20, 0])
         axs[1].set_ylim([-1.0, 0.0])
-    elif matter.lower() == 'esym':
-        axs[0].set_ylabel(r'$E_\text{sym}$',fontsize='14')
-        axs[1].set_ylabel(r'$E_\text{sym}/E_\text{sym,NRFFG}$',fontsize='14')
+    elif matter.lower() == "esym":
+        axs[0].set_ylabel(r"$E_\text{sym}$", fontsize="14")
+        axs[1].set_ylabel(r"$E_\text{sym}/E_\text{sym}^\text{NR FFG}$", fontsize="14")
         axs[0].set_ylim([0, 50])
         axs[1].set_ylim([1.5, 2.8])
     #
     for model in models:
         #
-        mic = nuda.matter.setupMicro( model = model )
-        if nuda.env.verb_output: mic.print_outputs( )
-        if mic.nm_e2a is not None: 
-            print('model:',model)
-            if matter.lower() == 'nm':
-                axs[0].errorbar( mic.nm_den, mic.nm_e2a, yerr= mic.nm_e2a_err, linestyle=mic.linestyle, label=mic.label, errorevery=mic.every )
-                axs[1].errorbar( mic.nm_den, mic.nm_e2a/nuda.effg_nr(mic.nm_kfn), yerr= mic.nm_e2a_err/nuda.effg_nr(mic.nm_kfn), linestyle=mic.linestyle, label=mic.label, errorevery=mic.every )
-            elif matter.lower() == 'sm':
-                axs[0].errorbar( mic.sm_den, mic.sm_e2a, yerr= mic.sm_e2a_err, linestyle=mic.linestyle, label=mic.label, errorevery=mic.every )
-                axs[1].errorbar( mic.sm_den, mic.sm_e2a/nuda.effg_nr(mic.sm_kfn), yerr= mic.sm_e2a_err/nuda.effg_nr(mic.sm_kfn), linestyle=mic.linestyle, label=mic.label, errorevery=mic.every )
-            elif matter.lower() == 'esym':
-                esym = nuda.matter.setupMicroEsym( model = model )
-                axs[0].errorbar( esym.den, esym.esym, yerr= esym.esym_err, linestyle=esym.linestyle, label=esym.label, errorevery=esym.every )
-                axs[1].errorbar( esym.den, esym.esym/nuda.esymffg_nr(esym.kf), yerr= esym.esym_err/nuda.esymffg_nr(esym.kf), linestyle=esym.linestyle, label=esym.label, errorevery=esym.every )
+        mic = nuda.matter.setupMicro(model=model)
+        if nuda.env.verb_output:
+            mic.print_outputs()
+        #
+        if mic.nm_e2a_int is not None:
+            print("model:", model)
+            if matter.lower() == "nm":
+                axs[0].errorbar(
+                    mic.nm_den,
+                    mic.nm_e2a_int,
+                    yerr=mic.nm_e2a_err,
+                    linestyle=mic.linestyle,
+                    label=mic.label,
+                    errorevery=mic.every,
+                )
+                axs[1].errorbar(
+                    mic.nm_den,
+                    mic.nm_e2a_int / nuda.effg_nr(mic.nm_kfn),
+                    yerr=mic.nm_e2a_err / nuda.effg_nr(mic.nm_kfn),
+                    linestyle=mic.linestyle,
+                    label=mic.label,
+                    errorevery=mic.every,
+                )
+            elif matter.lower() == "sm":
+                axs[0].errorbar(
+                    mic.sm_den,
+                    mic.sm_e2a_int,
+                    yerr=mic.sm_e2a_err,
+                    linestyle=mic.linestyle,
+                    label=mic.label,
+                    errorevery=mic.every,
+                )
+                axs[1].errorbar(
+                    mic.sm_den,
+                    mic.sm_e2a_int / nuda.effg_nr(mic.sm_kfn),
+                    yerr=mic.sm_e2a_err / nuda.effg_nr(mic.sm_kfn),
+                    linestyle=mic.linestyle,
+                    label=mic.label,
+                    errorevery=mic.every,
+                )
+            elif matter.lower() == "esym":
+                esym = nuda.matter.setupMicroEsym(model=model)
+                axs[0].errorbar(
+                    esym.den,
+                    esym.esym,
+                    yerr=esym.esym_err,
+                    linestyle=esym.linestyle,
+                    label=esym.label,
+                    errorevery=esym.every,
+                )
+                axs[1].errorbar(
+                    esym.den,
+                    esym.esym / nuda.esymffg_nr(esym.kf),
+                    yerr=esym.esym_err / nuda.esymffg_nr(esym.kf),
+                    linestyle=esym.linestyle,
+                    label=esym.label,
+                    errorevery=esym.every,
+                )
     #
-    band = nuda.matter.setupMicroBand( models, den=den, matter=matter )
+    band = nuda.matter.setupMicroBand(models, den=den, matter=matter)
     #
-    axs[0].fill_between( band.den, y1=band.e2a-band.e2a_std, y2=band.e2a+band.e2a_std, color=band.color, alpha=band.alpha )
-    if matter.lower() == 'nm':
-        axs[1].fill_between( band.den, y1=(band.e2a-band.e2a_std)/nuda.effg_nr(band.kfn), y2=(band.e2a+band.e2a_std)/nuda.effg_nr(band.kfn), color=band.color, alpha=band.alpha )
-    elif matter.lower() == 'sm':
-        axs[1].fill_between( band.den, y1=(band.e2a-band.e2a_std)/nuda.effg_nr(band.kf), y2=(band.e2a+band.e2a_std)/nuda.effg_nr(band.kf), color=band.color, alpha=band.alpha )
-    elif matter.lower() == 'esym':
-        axs[1].fill_between( band.den, y1=(band.e2a-band.e2a_std)/nuda.esymffg_nr(band.kf), y2=(band.e2a+band.e2a_std)/nuda.esymffg_nr(band.kf), color=band.color, alpha=band.alpha )
+    axs[0].fill_between(
+        band.den,
+        y1=band.e2a_int - band.e2a_std,
+        y2=band.e2a_int + band.e2a_std,
+        color=band.color,
+        alpha=band.alpha,
+    )
+    if matter.lower() == "nm":
+        axs[1].fill_between(
+            band.den,
+            y1=(band.e2a_int - band.e2a_std) / nuda.effg_nr(band.kfn),
+            y2=(band.e2a_int + band.e2a_std) / nuda.effg_nr(band.kfn),
+            color=band.color,
+            alpha=band.alpha,
+        )
+    elif matter.lower() == "sm":
+        axs[1].fill_between(
+            band.den,
+            y1=(band.e2a_int - band.e2a_std) / nuda.effg_nr(band.kf),
+            y2=(band.e2a_int + band.e2a_std) / nuda.effg_nr(band.kf),
+            color=band.color,
+            alpha=band.alpha,
+        )
+    elif matter.lower() == "esym":
+        axs[1].fill_between(
+            band.den,
+            y1=(band.e2a_int - band.e2a_std) / nuda.esymffg_nr(band.kf),
+            y2=(band.e2a_int + band.e2a_std) / nuda.esymffg_nr(band.kf),
+            color=band.color,
+            alpha=band.alpha,
+        )
     #
-    if matter.lower() == 'nm':
-        axs[1].legend(loc='upper left',fontsize='12', ncol=1)
-    elif matter.lower() == 'sm':
-        axs[1].legend(loc='upper left',fontsize='12', ncol=1)
-    elif matter.lower() == 'esym':
-        axs[0].legend(loc='upper left',fontsize='12', ncol=1)
+    if matter.lower() == "nm":
+        axs[1].legend(loc="upper left", fontsize="12", ncol=1)
+    elif matter.lower() == "sm":
+        axs[1].legend(loc="upper left", fontsize="12", ncol=1)
+    elif matter.lower() == "esym":
+        axs[0].legend(loc="upper left", fontsize="12", ncol=1)
     #
     if pname is not None:
-    	plt.savefig(pname, dpi=200)
-    	plt.close()
+        plt.savefig(pname, dpi=200)
+        plt.close()