add_assumed_redshift_to_3FHL.py 文件源码

def get_blazar_redshifts(blazar_type):
    table = Table.read(filename, hdu='LAT_Point_Source_Catalog')
    known_redshift_mask = np.isfinite(table['Redshift'])
    known_redshift_table = table[known_redshift_mask]
    if blazar_type == "bll":
        class_1 = known_redshift_table['CLASS'] == "bll    "
        class_2 = known_redshift_table['CLASS'] == "BLL    "
    if blazar_type == "fsrq":
        class_1 = known_redshift_table['CLASS'] == "fsrq   "
        class_2 = known_redshift_table['CLASS'] == "FSRQ   "
    if blazar_type == "bcu":
        class_1 = known_redshift_table['CLASS'] == "bcu    "
        class_2 = known_redshift_table['CLASS'] == "BCU    "
    class_type_mask = np.logical_or.reduce((class_1, class_2))
    sub_table = known_redshift_table[class_type_mask]
    return sub_table["Redshift"]