nugrid_set.py 文件源码

def find_most_abundant(self,coords=[],cycle=[]):

        minis=[]
        isotopes=[]
        abus=[]
                for i in range(len(self.runs_H5_out)):
                        sefiles=se(self.runs_H5_out[i])
                        mass_ini=sefiles.get("mini")
                        z=sefiles.get("zini")
            if cycle[i]==-1:
                cyc=sefiles.se.cycles[-1]
            else:
                cyc=cycle[i]
            coord=coords[i]
            mass=sefiles.get(cyc,'mass')
            mass_idx=min(range(len(mass)), key=lambda i: abs(mass[i]-coord))
            abu=sefiles.get(cyc,'iso_massf')[mass_idx]
            idx=list(abu).index(max(abu))
            print 'at mass coord',coord,': ',sefiles.se.isotopes[idx],'with ',max(abu)
            minis.append(mass_ini)
            isotopes.append(sefiles.se.isotopes[idx])
            abus.append(max(abu))

        for k in range(len(minis)):
            print 'M=',minis[k],'Z=',z,'at mass coord',coords[k],': ',isotopes[k],'with ',abus[k]