astrom_intra.py 文件源码

def get_mags(self, TT, apply, magcol='mag1', magerrcol='mag1_err'):
        alldd = []
        allmagI = []
        allmagJ = []
        allmagerrI = []
        allmagerrJ = []
        for ei in range(len(self.edges)):
            good = self.edge_good(ei)
            ii,jj = self.edge_ij(ei)
            I,J = self.edge_IJ(ei)
            (r,d, dra,ddec) = self.get_edge_dradec_arcsec(ei, corrected=apply,
                                                          goodonly=True)
            dd = np.sqrt(dra**2 + ddec**2)
            alldd.append(dd)
            # 'I' is an int array
            # 'good' is a bool array of size == size(I)
            # number of True elements in 'good' == len(dra)
            magI = TT[ii][I][good].get(magcol)
            magJ = TT[jj][J][good].get(magcol)
            allmagI.append(magI)
            allmagJ.append(magJ)
            if magerrcol in TT[ii].columns():
                allmagerrI.append(TT[ii].get(magerrcol)[I][good])
                allmagerrJ.append(TT[jj].get(magerrcol)[J][good])

        alldd = np.hstack(alldd)
        allmagI = np.hstack(allmagI)
        allmagJ = np.hstack(allmagJ)
        allmagerrI = np.hstack(allmagerrI)
        allmagerrJ = np.hstack(allmagerrJ)
        return (allmagI, allmagJ, alldd, magcol, allmagerrI, allmagerrJ)