nugrid_set.py 文件源码

def plot_final_surf_abu(self,yaxis='[Ba/Eu]',color='r',marker='o',linestyle='-',iniabupath='/astro/critter/critter/PPN/forum.astro.keele.ac.uk/frames/mppnp/USEEPP/iniab2.0E-02GN93.ppn'):


        elem1=yaxis.split('/')[0].split('[')[1]
        elem2=yaxis.split('/')[1].split(']')[0]

        elem1_ini=get_ini_elem_abu(element=elem1,iniabupath=iniabupath)
        elem2_ini=get_ini_elem_abu(element=elem2,iniabupath=iniabupath)

        spec_values=[]
        masses=[]
        for i in range(len(self.runs_H5_surf)):

            sefiles=se(self.runs_H5_surf[i])
            abu=[]
            abu1=self.get_elem_abu(cycles=-1,element=elem1,sefiles=sefiles)[0]  
            abu2=self.get_elem_abu(cycles=-1,element=elem2,sefiles=sefiles)[0]
            spec_values.append(np.log10(abu1/elem1_ini * elem2_ini/abu2))
                        print abu1,abu2
                        print elem1_ini,elem2_ini

            mass=sefiles.get("mini")
                        z=sefiles.get("zini")
            masses.append(mass)
                        legend=str(mass)+"M$_{\odot}$ Z= "+str(z)
        print spec_values
        print 'for ',elem1,elem2
        plt.figure(elem1+'_'+elem2)#+', Z='+str(z))
        plt.plot(masses,spec_values,marker=marker,color=color,linestyle=linestyle,label='Z='+str(z))
        plt.ylabel(yaxis)
        plt.xlabel('M/M$_{\odot}$')
        plt.legend(loc=1)