gaia_lbol.py

from astropy.io import ascii
import astropy.units as u
from astropy.coordinates import SkyCoord
from astroquery.gaia import Gaia


def find_one(ra, dec, mag):
    coord = SkyCoord(ra=ra, dec=dec, unit=(u.deg, u.deg))
    radius = 5 * u.arcsec
    j = Gaia.cone_search_async(coord, radius)
    r = j.get_results()

    # Restrict the magnitude
    r = r[abs(r["phot_g_mean_mag"] - mag) < 1.0]

    if not len(r):
        return float("nan")

    # Select the closest target
    r = r[0]
    return float(r["lum_val"])


data = ascii.read("https://arxiv.org/src/1501.01635v3/anc/Tables5_6_7.txt")
with open("lbol.csv", "w") as f:
    for row in data:
        f.write(
            f'{row["Name"]},{find_one(row["RAdeg"], row["DEdeg"], row["Gaiamag"])}\n'
        )