Main_script.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "import csv\n",
    "from astropy.io import fits\n",
    "import numpy as np\n",
    "import sys\n",
    "import math\n",
    "import re\n",
    "import matplotlib.pyplot as plt\n",
    "import matplotlib.path as path\n",
    "import matplotlib.patches as patch\n",
    "import random\n",
    "from scipy.stats import gaussian_kde\n",
    "from scipy.interpolate import interpn\n",
    "import matplotlib.cm as cm"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def vlm_object_probability(file,model,output_name):\n",
    "    \n",
    "    '''This is the main program used to find the low-mass objects. It takes as input a file of sources with ZYJHK\n",
    "    magnitues, and a model with 1 and 5 Myr lines. It is pretty specific to the ones I used, so I would suggest\n",
    "    starting with those if you are going to make modifications. It takes the input data, makes a poly-fit error,\n",
    "    and applies that to the model, along with Av for the extincted side. Then it uses those borders to flag sources\n",
    "    as being inside. Then it sums all those flags, and puts the results into columns, which are then aligned into\n",
    "    a single fits file. This verison returns a full list of all objects.'''\n",
    "    \n",
    "    RA_dms = []\n",
    "    Dec_dms = []\n",
    "    RA_Rad = []\n",
    "    Dec_Rad = []\n",
    "    RA_ICRS = []\n",
    "    Dec_ICRS = []\n",
    "    zmag = []\n",
    "    ymag = []\n",
    "    jmag = []\n",
    "    hmag = []\n",
    "    kmag = []\n",
    "    zobsmag = []\n",
    "    yobsmag = []\n",
    "    jobsmag = []\n",
    "    hobsmag = []\n",
    "    kobsmag = []\n",
    "    zaper = []\n",
    "    yaper = []\n",
    "    japer = []\n",
    "    haper = []\n",
    "    kaper = []\n",
    "    zapererr = []\n",
    "    yapererr = []\n",
    "    japererr = []\n",
    "    hapererr = []\n",
    "    kapererr = []\n",
    "    \n",
    "    intest = [[[[[0 for z in range(5)] for y in range(4)] for x in range(2)] for w in range(2)] for v in range(2)]\n",
    "    \n",
    "    hdul = fits.open(file)\n",
    "    allpoints = list(range(len(hdul[1].data)))\n",
    "    \n",
    "    Filters = [0 for z in allpoints]\n",
    "    YND = []\n",
    "    YNC = []\n",
    "    YFD = []\n",
    "    YFC = []\n",
    "    OND = []\n",
    "    ONC = []\n",
    "    OFD = []\n",
    "    OFC = []\n",
    "    ZY = []\n",
    "    ZJ = []\n",
    "    ZH = []\n",
    "    ZK = []\n",
    "    YJ = []\n",
    "    YH = []\n",
    "    YK = []\n",
    "    JH = []\n",
    "    JK = []\n",
    "    HK = []\n",
    "    Max = [0 for z in allpoints]\n",
    "    Total = [0 for z in allpoints]\n",
    "    Percentage = [0 for z in allpoints]\n",
    "    Y = [0 for z in allpoints]\n",
    "    O = [0 for z in allpoints]\n",
    "    N = [0 for z in allpoints]\n",
    "    F = [0 for z in allpoints]\n",
    "    D = [0 for z in allpoints]\n",
    "    C = [0 for z in allpoints]\n",
    "    YNDperc = [0 for z in allpoints]\n",
    "    YNCperc = [0 for z in allpoints]\n",
    "    YFDperc = [0 for z in allpoints]\n",
    "    YFCperc = [0 for z in allpoints]\n",
    "    ONDperc = [0 for z in allpoints]\n",
    "    ONCperc = [0 for z in allpoints]\n",
    "    OFDperc = [0 for z in allpoints]\n",
    "    OFCperc = [0 for z in allpoints]\n",
    "    Max_Perc = [0 for z in allpoints]\n",
    "\n",
    "    ra_dms = hdul[1].data['RA_dms']\n",
    "    dec_dms = hdul[1].data['Dec_dms']\n",
    "    ra_rad = hdul[1].data['RA_Rad']\n",
    "    dec_rad = hdul[1].data['Dec_Rad']\n",
    "    ra_icrs = hdul[1].data['RA_ICRS']\n",
    "    dec_icrs = hdul[1].data['Dec_ICRS']\n",
    "    mag_z = hdul[1].data['Mag_Z']\n",
    "    obs_mag_z = hdul[1].data['Obs_Mag_Z']\n",
    "    aper_z = hdul[1].data['Aper_Z']\n",
    "    aper_z_err = hdul[1].data['Aper_Z_err']\n",
    "    mag_y = hdul[1].data['Mag_Y']\n",
    "    obs_mag_y = hdul[1].data['Obs_Mag_Y']\n",
    "    aper_y = hdul[1].data['Aper_Y']\n",
    "    aper_y_err = hdul[1].data['Aper_Y_err']\n",
    "    mag_j = hdul[1].data['Mag_J']\n",
    "    obs_mag_j = hdul[1].data['Obs_Mag_J']\n",
    "    aper_j = hdul[1].data['Aper_J']\n",
    "    aper_j_err = hdul[1].data['Aper_J_err']\n",
    "    mag_h = hdul[1].data['Mag_H']\n",
    "    obs_mag_h = hdul[1].data['Obs_Mag_H']\n",
    "    aper_h = hdul[1].data['Aper_H']\n",
    "    aper_h_err = hdul[1].data['Aper_H_err']\n",
    "    mag_k = hdul[1].data['Mag_K']\n",
    "    obs_mag_k = hdul[1].data['Obs_Mag_K']\n",
    "    aper_k = hdul[1].data['Aper_K']\n",
    "    aper_k_err = hdul[1].data['Aper_K_err']\n",
    "    \n",
    "    err_z = aper_z_err/(aper_z)\n",
    "    err_y = aper_y_err/(aper_y)\n",
    "    err_j = aper_j_err/(aper_j)\n",
    "    err_h = aper_h_err/(aper_h)\n",
    "    err_k = aper_k_err/(aper_k)\n",
    "    \n",
    "    model = fits.open('model_iso.fits')\n",
    "\n",
    "    mass1 = model[1].data['Mass'][0:28]\n",
    "    teff1 = model[1].data['Teff'][0:28]\n",
    "    zmd1 = model[1].data['ZmagD'][0:28]\n",
    "    ymd1 = model[1].data['YmagD'][0:28]\n",
    "    jmd1 = model[1].data['JmagD'][0:28]\n",
    "    hmd1 = model[1].data['HmagD'][0:28]\n",
    "    kmd1 = model[1].data['KsmagD'][0:28]\n",
    "    zmc1 = model[1].data['ZmagC'][0:28]\n",
    "    ymc1 = model[1].data['YmagC'][0:28]\n",
    "    jmc1 = model[1].data['JmagC'][0:28]\n",
    "    hmc1 = model[1].data['HmagC'][0:28]\n",
    "    kmc1 = model[1].data['KsmagC'][0:28]\n",
    "    \n",
    "    mass5 = model[1].data['Mass'][28:57]\n",
    "    teff5 = model[1].data['Teff'][28:57]\n",
    "    zmd5 = model[1].data['ZmagD'][28:57]\n",
    "    ymd5 = model[1].data['YmagD'][28:57]\n",
    "    jmd5 = model[1].data['JmagD'][28:57]\n",
    "    hmd5 = model[1].data['HmagD'][28:57]\n",
    "    kmd5 = model[1].data['KsmagD'][28:57]\n",
    "    zmc5 = model[1].data['ZmagC'][28:57]\n",
    "    ymc5 = model[1].data['YmagC'][28:57]\n",
    "    jmc5 = model[1].data['JmagC'][28:57]\n",
    "    hmc5 = model[1].data['HmagC'][28:57]\n",
    "    kmc5 = model[1].data['KsmagC'][28:57]\n",
    "\n",
    "    massy_fit = np.polyfit(teff1, mass1, 14)\n",
    "    massy_poly = np.poly1d(massy_fit)\n",
    "    zcy_fit = np.polyfit(teff1, zmc1, 14)\n",
    "    zcy_poly = np.poly1d(zcy_fit)\n",
    "    zdy_fit = np.polyfit(teff1, zmd1, 14)\n",
    "    zdy_poly = np.poly1d(zdy_fit)\n",
    "    ycy_fit = np.polyfit(teff1, ymc1, 14)\n",
    "    ycy_poly = np.poly1d(ycy_fit)\n",
    "    ydy_fit = np.polyfit(teff1, ymd1, 14)\n",
    "    ydy_poly = np.poly1d(ydy_fit)\n",
    "    jcy_fit = np.polyfit(teff1, jmc1, 14)\n",
    "    jcy_poly = np.poly1d(jcy_fit)\n",
    "    jdy_fit = np.polyfit(teff1, jmd1, 14)\n",
    "    jdy_poly = np.poly1d(jdy_fit)\n",
    "    hcy_fit = np.polyfit(teff1, hmc1, 14)\n",
    "    hcy_poly = np.poly1d(hcy_fit)\n",
    "    hdy_fit = np.polyfit(teff1, hmd1, 14)\n",
    "    hdy_poly = np.poly1d(hdy_fit)\n",
    "    kcy_fit = np.polyfit(teff1, kmc1, 14)\n",
    "    kcy_poly = np.poly1d(kcy_fit)\n",
    "    kdy_fit = np.polyfit(teff1, kmd1, 14)\n",
    "    kdy_poly = np.poly1d(kdy_fit)\n",
    "    \n",
    "    masso_fit = np.polyfit(teff5, mass5, 14)\n",
    "    masso_poly = np.poly1d(masso_fit)\n",
    "    zco_fit = np.polyfit(teff5, zmc5, 14)\n",
    "    zco_poly = np.poly1d(zco_fit)\n",
    "    zdo_fit = np.polyfit(teff5, zmd5, 14)\n",
    "    zdo_poly = np.poly1d(zdo_fit)\n",
    "    yco_fit = np.polyfit(teff5, ymc5, 14)\n",
    "    yco_poly = np.poly1d(yco_fit)\n",
    "    ydo_fit = np.polyfit(teff5, ymd5, 14)\n",
    "    ydo_poly = np.poly1d(ydo_fit)\n",
    "    jco_fit = np.polyfit(teff5, jmc5, 14)\n",
    "    jco_poly = np.poly1d(jco_fit)\n",
    "    jdo_fit = np.polyfit(teff5, jmd5, 14)\n",
    "    jdo_poly = np.poly1d(jdo_fit)\n",
    "    hco_fit = np.polyfit(teff5, hmc5, 14)\n",
    "    hco_poly = np.poly1d(hco_fit)\n",
    "    hdo_fit = np.polyfit(teff5, hmd5, 14)\n",
    "    hdo_poly = np.poly1d(hdo_fit)\n",
    "    kco_fit = np.polyfit(teff5, kmc5, 14)\n",
    "    kco_poly = np.poly1d(kco_fit)\n",
    "    kdo_fit = np.polyfit(teff5, kmd5, 14)\n",
    "    kdo_poly = np.poly1d(kdo_fit)\n",
    "    \n",
    "    idz = np.isfinite(mag_z) & np.isfinite(err_z)\n",
    "    zerr_fit = np.polyfit(mag_z[idz],err_z[idz],8)\n",
    "    zerr_poly = np.poly1d(zerr_fit)\n",
    "    idy = np.isfinite(mag_y) & np.isfinite(err_y)\n",
    "    yerr_fit = np.polyfit(mag_y[idy],err_y[idy],8)\n",
    "    yerr_poly = np.poly1d(yerr_fit)\n",
    "    idj = np.isfinite(mag_j) & np.isfinite(err_j)\n",
    "    jerr_fit = np.polyfit(mag_j[idj],err_j[idj],8)\n",
    "    jerr_poly = np.poly1d(jerr_fit)\n",
    "    idh = np.isfinite(mag_h) & np.isfinite(err_h)\n",
    "    herr_fit = np.polyfit(mag_h[idh],err_h[idh],8)\n",
    "    herr_poly = np.poly1d(herr_fit)\n",
    "    idk = np.isfinite(mag_k) & np.isfinite(err_k)\n",
    "    kerr_fit = np.polyfit(mag_k[idk],err_k[idk],8)\n",
    "    kerr_poly = np.poly1d(kerr_fit)\n",
    "    \n",
    "    bd_range_1myr = range(1300,2931)\n",
    "    full_rangey = range(1300,4000)\n",
    "    full_rangeo = range(1300,4000)\n",
    "    full_rangec = range(1300,4000)\n",
    "    full_ranged = range(1650,4000)\n",
    "    \n",
    "    mu_far = Distance(420)\n",
    "    mu_near = Distance(350)\n",
    "    Av = 1.1\n",
    "    \n",
    "    Az = 0.5006359319 * Av\n",
    "    Ay = 0.3913227425 * Av\n",
    "    Aj = 0.281344478 * Av\n",
    "    Ah = 0.1812818241 * Av\n",
    "    Ak = 0.1180678823 * Av\n",
    "    \n",
    "    Azy = Az - Ay\n",
    "    Azj = Az - Aj\n",
    "    Azh = Az - Ah\n",
    "    Azk = Az - Ak\n",
    "    Ayj = Ay - Aj\n",
    "    Ayh = Ay - Ah\n",
    "    Ayk = Ay - Ak\n",
    "    Ajh = Aj - Ah\n",
    "    Ajk = Aj - Ak\n",
    "    Ahk = Ah - Ak\n",
    "    \n",
    "    poly = [[[zdy_poly(full_ranged),zdo_poly(full_ranged)],[zcy_poly(full_rangec),zco_poly(full_rangec)]],\n",
    "            [[ydy_poly(full_ranged),ydo_poly(full_ranged)],[ycy_poly(full_rangec),yco_poly(full_rangec)]],\n",
    "            [[jdy_poly(full_ranged),jdo_poly(full_ranged)],[jcy_poly(full_rangec),jco_poly(full_rangec)]],\n",
    "            [[hdy_poly(full_ranged),hdo_poly(full_ranged)],[hcy_poly(full_rangec),hco_poly(full_rangec)]],\n",
    "            [[kdy_poly(full_ranged),kdo_poly(full_ranged)],[kcy_poly(full_rangec),kco_poly(full_rangec)]]]\n",
    "    \n",
    "    observed_data = [mag_z,mag_y,mag_j,mag_h,mag_k]\n",
    "    observed_array = np.array(observed_data)\n",
    "\n",
    "    ext = [[Az,Ay,Aj,Ah,Ak],[[0,Azy,Azj,Azh,Azk],[0,0,Ayj,Ayh,Ayk],[0,0,0,Ajh,Ajk],[0,0,0,0,Ahk]]]\n",
    "    \n",
    "    mu = [mu_near,mu_far]\n",
    "    \n",
    "    for alpha in range(2):\n",
    "        for beta in range(2):\n",
    "            for gamma in range(2):\n",
    "                for delta in range(4):\n",
    "                    for epsilon in range(delta+1,5):\n",
    "                        \n",
    "                        err_poly_max = [zerr_poly(poly[delta][gamma][beta] + mu[alpha] + 1*ext[0][delta]),\n",
    "                                    yerr_poly(poly[delta][gamma][beta] + mu[alpha] + 1*ext[0][delta]),\n",
    "                                    jerr_poly(poly[delta][gamma][beta] + mu[alpha] + 1*ext[0][delta]),\n",
    "                                    herr_poly(poly[delta][gamma][beta] + mu[alpha] + 1*ext[0][delta]),\n",
    "                                    kerr_poly(poly[delta][gamma][beta] + mu[alpha] + 1*ext[0][delta])]\n",
    "                        \n",
    "                        err_poly_max_ep = [zerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 1*ext[0][epsilon]),\n",
    "                                    yerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 1*ext[0][epsilon]),\n",
    "                                    jerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 1*ext[0][epsilon]),\n",
    "                                    herr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 1*ext[0][epsilon]),\n",
    "                                    kerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 1*ext[0][epsilon])]\n",
    "                        \n",
    "                        err_poly_no = [zerr_poly(poly[delta][gamma][beta] + mu[alpha] + 0*ext[0][delta]),\n",
    "                                    yerr_poly(poly[delta][gamma][beta] + mu[alpha] + 0*ext[0][delta]),\n",
    "                                    jerr_poly(poly[delta][gamma][beta] + mu[alpha] + 0*ext[0][delta]),\n",
    "                                    herr_poly(poly[delta][gamma][beta] + mu[alpha] + 0*ext[0][delta]),\n",
    "                                    kerr_poly(poly[delta][gamma][beta] + mu[alpha] + 0*ext[0][delta])]\n",
    "                        \n",
    "                        err_poly_no_ep = [zerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 0*ext[0][epsilon]),\n",
    "                                    yerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 0*ext[0][epsilon]),\n",
    "                                    jerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 0*ext[0][epsilon]),\n",
    "                                    herr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 0*ext[0][epsilon]),\n",
    "                                    kerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 0*ext[0][epsilon])]\n",
    "                            \n",
    "                        maxext = list(zip((poly[delta][gamma][beta] - \n",
    "                                           poly[epsilon][gamma][beta] + \n",
    "                                           (1*ext[1][delta][epsilon])) + \n",
    "                                          np.sqrt(err_poly_max[delta]**2 + err_poly_max_ep[epsilon]**2),\n",
    "                                          poly[delta][gamma][beta] + mu[alpha] + \n",
    "                                          (1*ext[0][delta]) - err_poly_max[delta]))\n",
    "                        noext = list(zip((poly[delta][gamma][beta] - \n",
    "                                          poly[epsilon][gamma][beta] + \n",
    "                                          (0*ext[1][delta][epsilon])) - \n",
    "                                         np.sqrt(err_poly_no[delta]**2 + err_poly_no_ep[epsilon]**2),\n",
    "                                         poly[delta][gamma][beta] + mu[alpha] + \n",
    "                                         (0*ext[0][delta]) + err_poly_no[delta]))\n",
    "                        \n",
    "                        sidepoints = maxext + noext[::-1]\n",
    "                        polypath = path.Path(sidepoints, closed=True)\n",
    "                        \n",
    "                        points = list(zip(observed_data[delta]-observed_data[epsilon],observed_data[delta]))\n",
    "                        intest[beta][alpha][gamma][delta][epsilon] = polypath.contains_points(points)\n",
    "    \n",
    "    for i in allpoints:\n",
    "        Filters[i] = np.sum(~np.isnan(observed_array[:,i]))\n",
    "        \n",
    "        outynd = []\n",
    "        outync = []\n",
    "        outyfd = []\n",
    "        outyfc = []\n",
    "        outond = []\n",
    "        outonc = []\n",
    "        outofd = []\n",
    "        outofc = []\n",
    "\n",
    "        for d in range(4):\n",
    "            for e in range(d+1,5):\n",
    "                outynd.append(intest[0][0][0][d][e][i])\n",
    "                outync.append(intest[0][0][1][d][e][i])\n",
    "                outyfd.append(intest[0][1][0][d][e][i])\n",
    "                outyfc.append(intest[0][1][1][d][e][i])\n",
    "                outond.append(intest[1][0][0][d][e][i])\n",
    "                outonc.append(intest[1][0][1][d][e][i])\n",
    "                outofd.append(intest[1][1][0][d][e][i])\n",
    "                outofc.append(intest[1][1][1][d][e][i])\n",
    "        YND.append(np.sum(outynd))\n",
    "        YNC.append(np.sum(outync))\n",
    "        YFD.append(np.sum(outyfd))\n",
    "        YFC.append(np.sum(outyfc))\n",
    "        OND.append(np.sum(outond))\n",
    "        ONC.append(np.sum(outonc))\n",
    "        OFD.append(np.sum(outofd))\n",
    "        OFC.append(np.sum(outofc))\n",
    "\n",
    "        outzy = []\n",
    "        outzj = []\n",
    "        outzh = []\n",
    "        outzk = []\n",
    "        outyj = []\n",
    "        outyh = []\n",
    "        outyk = []\n",
    "        outjh = []\n",
    "        outjk = []\n",
    "        outhk = []\n",
    "\n",
    "        for a in range(2):\n",
    "            for b in range(2):\n",
    "                for c in range(2):\n",
    "                    outzy.append(intest[a][b][c][0][1][i])\n",
    "                    outzj.append(intest[a][b][c][0][2][i])\n",
    "                    outzh.append(intest[a][b][c][0][3][i])\n",
    "                    outzk.append(intest[a][b][c][0][4][i])\n",
    "                    outyj.append(intest[a][b][c][1][2][i])\n",
    "                    outyh.append(intest[a][b][c][1][3][i])\n",
    "                    outyk.append(intest[a][b][c][1][4][i])\n",
    "                    outjh.append(intest[a][b][c][2][3][i])\n",
    "                    outjk.append(intest[a][b][c][2][4][i])\n",
    "                    outhk.append(intest[a][b][c][3][4][i])\n",
    "        ZY.append(np.sum(outzy))\n",
    "        ZJ.append(np.sum(outzj))\n",
    "        ZH.append(np.sum(outzh))\n",
    "        ZK.append(np.sum(outzk))\n",
    "        YJ.append(np.sum(outyj))\n",
    "        YH.append(np.sum(outyh))\n",
    "        YK.append(np.sum(outyk))\n",
    "        JH.append(np.sum(outjh))\n",
    "        JK.append(np.sum(outjk))\n",
    "        HK.append(np.sum(outhk))\n",
    "        \n",
    "        if Filters[i] == 2:\n",
    "            Max[i] = 1*8\n",
    "        if Filters[i] == 3:\n",
    "            Max[i] = 3*8\n",
    "        if Filters[i] == 4:\n",
    "            Max[i] = 6*8\n",
    "        if Filters[i] == 5:\n",
    "            Max[i] = 10*8\n",
    "           \n",
    "        Y[i] = np.sum(YND[i] + YNC[i] + YFD[i] + YFC[i])\n",
    "        O[i] = np.sum(OND[i] + ONC[i] + OFD[i] + OFC[i])\n",
    "        N[i] = np.sum(YND[i] + YNC[i] + OND[i] + ONC[i])\n",
    "        F[i] = np.sum(YFD[i] + YFC[i] + OFD[i] + OFC[i])\n",
    "        D[i] = np.sum(YND[i] + OND[i] + YFD[i] + OFD[i])\n",
    "        C[i] = np.sum(YNC[i] + ONC[i] + YFC[i] + OFC[i])\n",
    "    \n",
    "        Total[i] = sum([YND[i],YNC[i],YFD[i],YFC[i],OND[i],ONC[i],OFD[i],OFC[i]])\n",
    "        YNDperc[i] = YND[i]/(Max[i]/8)\n",
    "        YNCperc[i] = YNC[i]/(Max[i]/8)\n",
    "        YFDperc[i] = YFD[i]/(Max[i]/8)\n",
    "        YFCperc[i] = YFC[i]/(Max[i]/8)\n",
    "        ONDperc[i] = OND[i]/(Max[i]/8)\n",
    "        ONCperc[i] = ONC[i]/(Max[i]/8)\n",
    "        OFDperc[i] = OFD[i]/(Max[i]/8)\n",
    "        OFCperc[i] = OFC[i]/(Max[i]/8)\n",
    "        Percentage[i] = Total[i]/Max[i]\n",
    "        Max_Perc[i] = max(YNDperc[i],YNCperc[i],YFDperc[i],YFCperc[i],\n",
    "                          ONDperc[i],ONCperc[i],OFDperc[i],OFCperc[i],\n",
    "                          Percentage[i])\n",
    "    \n",
    "    YND_zy = intest[0][0][0][0][1]\n",
    "    YND_zj = intest[0][0][0][0][2]\n",
    "    YND_zh = intest[0][0][0][0][3]\n",
    "    YND_zk = intest[0][0][0][0][4]\n",
    "    YND_yj = intest[0][0][0][1][2]\n",
    "    YND_yh = intest[0][0][0][1][3]\n",
    "    YND_yk = intest[0][0][0][1][4]\n",
    "    YND_jh = intest[0][0][0][2][3]\n",
    "    YND_jk = intest[0][0][0][2][4]\n",
    "    YND_hk = intest[0][0][0][3][4]\n",
    "    YNC_zy = intest[0][0][1][0][1]\n",
    "    YNC_zj = intest[0][0][1][0][2]\n",
    "    YNC_zh = intest[0][0][1][0][3]\n",
    "    YNC_zk = intest[0][0][1][0][4]\n",
    "    YNC_yj = intest[0][0][1][1][2]\n",
    "    YNC_yh = intest[0][0][1][1][3]\n",
    "    YNC_yk = intest[0][0][1][1][4]\n",
    "    YNC_jh = intest[0][0][1][2][3]\n",
    "    YNC_jk = intest[0][0][1][2][4]\n",
    "    YNC_hk = intest[0][0][1][3][4]\n",
    "    YFD_zy = intest[0][1][0][0][1]\n",
    "    YFD_zj = intest[0][1][0][0][2]\n",
    "    YFD_zh = intest[0][1][0][0][3]\n",
    "    YFD_zk = intest[0][1][0][0][4]\n",
    "    YFD_yj = intest[0][1][0][1][2]\n",
    "    YFD_yh = intest[0][1][0][1][3]\n",
    "    YFD_yk = intest[0][1][0][1][4]\n",
    "    YFD_jh = intest[0][1][0][2][3]\n",
    "    YFD_jk = intest[0][1][0][2][4]\n",
    "    YFD_hk = intest[0][1][0][3][4]\n",
    "    YFC_zy = intest[0][1][1][0][1]\n",
    "    YFC_zj = intest[0][1][1][0][2]\n",
    "    YFC_zh = intest[0][1][1][0][3]\n",
    "    YFC_zk = intest[0][1][1][0][4]\n",
    "    YFC_yj = intest[0][1][1][1][2]\n",
    "    YFC_yh = intest[0][1][1][1][3]\n",
    "    YFC_yk = intest[0][1][1][1][4]\n",
    "    YFC_jh = intest[0][1][1][2][3]\n",
    "    YFC_jk = intest[0][1][1][2][4]\n",
    "    YFC_hk = intest[0][1][1][3][4]\n",
    "    OND_zy = intest[1][0][0][0][1]\n",
    "    OND_zj = intest[1][0][0][0][2]\n",
    "    OND_zh = intest[1][0][0][0][3]\n",
    "    OND_zk = intest[1][0][0][0][4]\n",
    "    OND_yj = intest[1][0][0][1][2]\n",
    "    OND_yh = intest[1][0][0][1][3]\n",
    "    OND_yk = intest[1][0][0][1][4]\n",
    "    OND_jh = intest[1][0][0][2][3]\n",
    "    OND_jk = intest[1][0][0][2][4]\n",
    "    OND_hk = intest[1][0][0][3][4]\n",
    "    ONC_zy = intest[1][0][1][0][1]\n",
    "    ONC_zj = intest[1][0][1][0][2]\n",
    "    ONC_zh = intest[1][0][1][0][3]\n",
    "    ONC_zk = intest[1][0][1][0][4]\n",
    "    ONC_yj = intest[1][0][1][1][2]\n",
    "    ONC_yh = intest[1][0][1][1][3]\n",
    "    ONC_yk = intest[1][0][1][1][4]\n",
    "    ONC_jh = intest[1][0][1][2][3]\n",
    "    ONC_jk = intest[1][0][1][2][4]\n",
    "    ONC_hk = intest[1][0][1][3][4]\n",
    "    OFD_zy = intest[1][1][0][0][1]\n",
    "    OFD_zj = intest[1][1][0][0][2]\n",
    "    OFD_zh = intest[1][1][0][0][3]\n",
    "    OFD_zk = intest[1][1][0][0][4]\n",
    "    OFD_yj = intest[1][1][0][1][2]\n",
    "    OFD_yh = intest[1][1][0][1][3]\n",
    "    OFD_yk = intest[1][1][0][1][4]\n",
    "    OFD_jh = intest[1][1][0][2][3]\n",
    "    OFD_jk = intest[1][1][0][2][4]\n",
    "    OFD_hk = intest[1][1][0][3][4]\n",
    "    OFC_zy = intest[1][1][1][0][1]\n",
    "    OFC_zj = intest[1][1][1][0][2]\n",
    "    OFC_zh = intest[1][1][1][0][3]\n",
    "    OFC_zk = intest[1][1][1][0][4]\n",
    "    OFC_yj = intest[1][1][1][1][2]\n",
    "    OFC_yh = intest[1][1][1][1][3]\n",
    "    OFC_yk = intest[1][1][1][1][4]\n",
    "    OFC_jh = intest[1][1][1][2][3]\n",
    "    OFC_jk = intest[1][1][1][2][4]\n",
    "    OFC_hk = intest[1][1][1][3][4]\n",
    "    \n",
    "    output_fits = output_name + '.fits'\n",
    "    output_reg = output_name + '.reg'\n",
    "    \n",
    "    for j in allpoints:\n",
    "        RA_dms.append(hdul[1].data['RA_dms'][j])\n",
    "        Dec_dms.append(hdul[1].data['Dec_dms'][j])\n",
    "        RA_Rad.append(hdul[1].data['RA_Rad'][j])\n",
    "        Dec_Rad.append(hdul[1].data['Dec_Rad'][j])\n",
    "        RA_ICRS.append(hdul[1].data['RA_ICRS'][j])\n",
    "        Dec_ICRS.append(hdul[1].data['Dec_ICRS'][j])\n",
    "        zmag.append(hdul[1].data['Mag_Z'][j])\n",
    "        ymag.append(hdul[1].data['Mag_Y'][j])\n",
    "        jmag.append(hdul[1].data['Mag_J'][j])\n",
    "        hmag.append(hdul[1].data['Mag_H'][j])\n",
    "        kmag.append(hdul[1].data['Mag_K'][j])\n",
    "        zobsmag.append(hdul[1].data['Obs_Mag_Z'][j])\n",
    "        yobsmag.append(hdul[1].data['Obs_Mag_Y'][j])\n",
    "        jobsmag.append(hdul[1].data['Obs_Mag_J'][j])\n",
    "        hobsmag.append(hdul[1].data['Obs_Mag_H'][j])\n",
    "        kobsmag.append(hdul[1].data['Obs_Mag_K'][j])\n",
    "        zaper.append(hdul[1].data['Aper_Z'][j])\n",
    "        yaper.append(hdul[1].data['Aper_Y'][j])\n",
    "        japer.append(hdul[1].data['Aper_J'][j])\n",
    "        haper.append(hdul[1].data['Aper_H'][j])\n",
    "        kaper.append(hdul[1].data['Aper_K'][j])\n",
    "        zapererr.append(hdul[1].data['Aper_Z_err'][j])\n",
    "        yapererr.append(hdul[1].data['Aper_Y_err'][j])\n",
    "        japererr.append(hdul[1].data['Aper_J_err'][j])\n",
    "        hapererr.append(hdul[1].data['Aper_H_err'][j])\n",
    "        kapererr.append(hdul[1].data['Aper_K_err'][j])\n",
    "    \n",
    "    col1 = fits.Column(name='RA_dms',format='11A',unit='dms',array=RA_dms)\n",
    "    col2 = fits.Column(name='Dec_dms',format='11A',unit='dms',array=Dec_dms)\n",
    "    col3 = fits.Column(name='RA_Rad',format='D',unit='radians',array=RA_Rad)\n",
    "    col4 = fits.Column(name='Dec_Rad',format='D',unit='radians',array=Dec_Rad)\n",
    "    col3a = fits.Column(name='RA_ICRS',format='D',unit='degree',array=RA_ICRS)\n",
    "    col4a = fits.Column(name='Dec_ICRS',format='D',unit='degree',array=Dec_ICRS)\n",
    "    col5 = fits.Column(name='Mag_Z',format='D',unit='mag',array=zmag)\n",
    "    col6 = fits.Column(name='Obs_Mag_Z',format='D',unit='mag',array=zobsmag)\n",
    "    col7 = fits.Column(name='Aper_Z',format='E',unit='ADU',array=zaper)\n",
    "    col8 = fits.Column(name='Aper_Z_err',format='E',unit='ADU',array=zapererr)\n",
    "    col9 = fits.Column(name='Mag_Y',format='D',unit='mag',array=ymag)\n",
    "    col10 = fits.Column(name='Obs_Mag_Y',format='D',unit='mag',array=yobsmag)\n",
    "    col11 = fits.Column(name='Aper_Y',format='E',unit='ADU',array=yaper)\n",
    "    col12 = fits.Column(name='Aper_Y_err',format='E',unit='ADU',array=yapererr)\n",
    "    col13 = fits.Column(name='Mag_J',format='D',unit='mag',array=jmag)\n",
    "    col14 = fits.Column(name='Obs_Mag_J',format='D',unit='mag',array=jobsmag)\n",
    "    col15 = fits.Column(name='Aper_J',format='E',unit='ADU',array=japer)\n",
    "    col16 = fits.Column(name='Aper_J_err',format='E',unit='ADU',array=japererr)\n",
    "    col17 = fits.Column(name='Mag_H',format='D',unit='mag',array=hmag)\n",
    "    col18 = fits.Column(name='Obs_Mag_H',format='D',unit='mag',array=hobsmag)\n",
    "    col19 = fits.Column(name='Aper_H',format='E',unit='ADU',array=haper)\n",
    "    col20 = fits.Column(name='Aper_H_err',format='E',unit='ADU',array=hapererr)\n",
    "    col21 = fits.Column(name='Mag_K',format='D',unit='mag',array=kmag)\n",
    "    col22 = fits.Column(name='Obs_Mag_K',format='D',unit='mag',array=kobsmag)\n",
    "    col23 = fits.Column(name='Aper_K',format='E',unit='ADU',array=kaper)\n",
    "    col24 = fits.Column(name='Aper_K_err',format='E',unit='ADU',array=kapererr)\n",
    "    col25 = fits.Column(name='YND_zy',format='11A',unit='Flag',array=YND_zy)\n",
    "    col26 = fits.Column(name='YND_zj',format='11A',unit='Flag',array=YND_zj)\n",
    "    col27 = fits.Column(name='YND_zh',format='11A',unit='Flag',array=YND_zh)\n",
    "    col28 = fits.Column(name='YND_zk',format='11A',unit='Flag',array=YND_zk)\n",
    "    col29 = fits.Column(name='YND_yj',format='11A',unit='Flag',array=YND_yj)\n",
    "    col30 = fits.Column(name='YND_yh',format='11A',unit='Flag',array=YND_yh)\n",
    "    col31 = fits.Column(name='YND_yk',format='11A',unit='Flag',array=YND_yk)\n",
    "    col32 = fits.Column(name='YND_jh',format='11A',unit='Flag',array=YND_jh)\n",
    "    col33 = fits.Column(name='YND_jk',format='11A',unit='Flag',array=YND_jk)\n",
    "    col34 = fits.Column(name='YND_hk',format='11A',unit='Flag',array=YND_hk)\n",
    "    col35 = fits.Column(name='YNC_zy',format='11A',unit='Flag',array=YNC_zy)\n",
    "    col36 = fits.Column(name='YNC_zj',format='11A',unit='Flag',array=YNC_zj)\n",
    "    col37 = fits.Column(name='YNC_zh',format='11A',unit='Flag',array=YNC_zh)\n",
    "    col38 = fits.Column(name='YNC_zk',format='11A',unit='Flag',array=YNC_zk)\n",
    "    col39 = fits.Column(name='YNC_yj',format='11A',unit='Flag',array=YNC_yj)\n",
    "    col40 = fits.Column(name='YNC_yh',format='11A',unit='Flag',array=YNC_yh)\n",
    "    col41 = fits.Column(name='YNC_yk',format='11A',unit='Flag',array=YNC_yk)\n",
    "    col42 = fits.Column(name='YNC_jh',format='11A',unit='Flag',array=YNC_jh)\n",
    "    col43 = fits.Column(name='YNC_jk',format='11A',unit='Flag',array=YNC_jk)\n",
    "    col44 = fits.Column(name='YNC_hk',format='11A',unit='Flag',array=YNC_hk)\n",
    "    col45 = fits.Column(name='YFD_zy',format='11A',unit='Flag',array=YFD_zy)\n",
    "    col46 = fits.Column(name='YFD_zj',format='11A',unit='Flag',array=YFD_zj)\n",
    "    col47 = fits.Column(name='YFD_zh',format='11A',unit='Flag',array=YFD_zh)\n",
    "    col48 = fits.Column(name='YFD_zk',format='11A',unit='Flag',array=YFD_zk)\n",
    "    col49 = fits.Column(name='YFD_yj',format='11A',unit='Flag',array=YFD_yj)\n",
    "    col50 = fits.Column(name='YFD_yh',format='11A',unit='Flag',array=YFD_yh)\n",
    "    col51 = fits.Column(name='YFD_yk',format='11A',unit='Flag',array=YFD_yk)\n",
    "    col52 = fits.Column(name='YFD_jh',format='11A',unit='Flag',array=YFD_jh)\n",
    "    col53 = fits.Column(name='YFD_jk',format='11A',unit='Flag',array=YFD_jk)\n",
    "    col54 = fits.Column(name='YFD_hk',format='11A',unit='Flag',array=YFD_hk)\n",
    "    col55 = fits.Column(name='YFC_zy',format='11A',unit='Flag',array=YFC_zy)\n",
    "    col56 = fits.Column(name='YFC_zj',format='11A',unit='Flag',array=YFC_zj)\n",
    "    col57 = fits.Column(name='YFC_zh',format='11A',unit='Flag',array=YFC_zh)\n",
    "    col58 = fits.Column(name='YFC_zk',format='11A',unit='Flag',array=YFC_zk)\n",
    "    col59 = fits.Column(name='YFC_yj',format='11A',unit='Flag',array=YFC_yj)\n",
    "    col60 = fits.Column(name='YFC_yh',format='11A',unit='Flag',array=YFC_yh)\n",
    "    col61 = fits.Column(name='YFC_yk',format='11A',unit='Flag',array=YFC_yk)\n",
    "    col62 = fits.Column(name='YFC_jh',format='11A',unit='Flag',array=YFC_jh)\n",
    "    col63 = fits.Column(name='YFC_jk',format='11A',unit='Flag',array=YFC_jk)\n",
    "    col64 = fits.Column(name='YFC_hk',format='11A',unit='Flag',array=YFC_hk)\n",
    "    col65 = fits.Column(name='OND_zy',format='11A',unit='Flag',array=OND_zy)\n",
    "    col66 = fits.Column(name='OND_zj',format='11A',unit='Flag',array=OND_zj)\n",
    "    col67 = fits.Column(name='OND_zh',format='11A',unit='Flag',array=OND_zh)\n",
    "    col68 = fits.Column(name='OND_zk',format='11A',unit='Flag',array=OND_zk)\n",
    "    col69 = fits.Column(name='OND_yj',format='11A',unit='Flag',array=OND_yj)\n",
    "    col70 = fits.Column(name='OND_yh',format='11A',unit='Flag',array=OND_yh)\n",
    "    col71 = fits.Column(name='OND_yk',format='11A',unit='Flag',array=OND_yk)\n",
    "    col72 = fits.Column(name='OND_jh',format='11A',unit='Flag',array=OND_jh)\n",
    "    col73 = fits.Column(name='OND_jk',format='11A',unit='Flag',array=OND_jk)\n",
    "    col74 = fits.Column(name='OND_hk',format='11A',unit='Flag',array=OND_hk)\n",
    "    col75 = fits.Column(name='ONC_zy',format='11A',unit='Flag',array=ONC_zy)\n",
    "    col76 = fits.Column(name='ONC_zj',format='11A',unit='Flag',array=ONC_zj)\n",
    "    col77 = fits.Column(name='ONC_zh',format='11A',unit='Flag',array=ONC_zh)\n",
    "    col78 = fits.Column(name='ONC_zk',format='11A',unit='Flag',array=ONC_zk)\n",
    "    col79 = fits.Column(name='ONC_yj',format='11A',unit='Flag',array=ONC_yj)\n",
    "    col80 = fits.Column(name='ONC_yh',format='11A',unit='Flag',array=ONC_yh)\n",
    "    col81 = fits.Column(name='ONC_yk',format='11A',unit='Flag',array=ONC_yk)\n",
    "    col82 = fits.Column(name='ONC_jh',format='11A',unit='Flag',array=ONC_jh)\n",
    "    col83 = fits.Column(name='ONC_jk',format='11A',unit='Flag',array=ONC_jk)\n",
    "    col84 = fits.Column(name='ONC_hk',format='11A',unit='Flag',array=ONC_hk)\n",
    "    col85 = fits.Column(name='OFD_zy',format='11A',unit='Flag',array=OFD_zy)\n",
    "    col86 = fits.Column(name='OFD_zj',format='11A',unit='Flag',array=OFD_zj)\n",
    "    col87 = fits.Column(name='OFD_zh',format='11A',unit='Flag',array=OFD_zh)\n",
    "    col88 = fits.Column(name='OFD_zk',format='11A',unit='Flag',array=OFD_zk)\n",
    "    col89 = fits.Column(name='OFD_yj',format='11A',unit='Flag',array=OFD_yj)\n",
    "    col90 = fits.Column(name='OFD_yh',format='11A',unit='Flag',array=OFD_yh)\n",
    "    col91 = fits.Column(name='OFD_yk',format='11A',unit='Flag',array=OFD_yk)\n",
    "    col92 = fits.Column(name='OFD_jh',format='11A',unit='Flag',array=OFD_jh)\n",
    "    col93 = fits.Column(name='OFD_jk',format='11A',unit='Flag',array=OFD_jk)\n",
    "    col94 = fits.Column(name='OFD_hk',format='11A',unit='Flag',array=OFD_hk)\n",
    "    col95 = fits.Column(name='OFC_zy',format='11A',unit='Flag',array=OFC_zy)\n",
    "    col96 = fits.Column(name='OFC_zj',format='11A',unit='Flag',array=OFC_zj)\n",
    "    col97 = fits.Column(name='OFC_zh',format='11A',unit='Flag',array=OFC_zh)\n",
    "    col98 = fits.Column(name='OFC_zk',format='11A',unit='Flag',array=OFC_zk)\n",
    "    col99 = fits.Column(name='OFC_yj',format='11A',unit='Flag',array=OFC_yj)\n",
    "    col100 = fits.Column(name='OFC_yh',format='11A',unit='Flag',array=OFC_yh)\n",
    "    col101 = fits.Column(name='OFC_yk',format='11A',unit='Flag',array=OFC_yk)\n",
    "    col102 = fits.Column(name='OFC_jh',format='11A',unit='Flag',array=OFC_jh)\n",
    "    col103 = fits.Column(name='OFC_jk',format='11A',unit='Flag',array=OFC_jk)\n",
    "    col104 = fits.Column(name='OFC_hk',format='11A',unit='Flag',array=OFC_hk)\n",
    "    col105 = fits.Column(name='YND Total',format='F',unit='Counts',array=YND)\n",
    "    col106 = fits.Column(name='YNC Total',format='F',unit='Counts',array=YNC)\n",
    "    col107 = fits.Column(name='YFD Total',format='F',unit='Counts',array=YFD)\n",
    "    col108 = fits.Column(name='YFC Total',format='F',unit='Counts',array=YFC)\n",
    "    col109 = fits.Column(name='OND Total',format='F',unit='Counts',array=OND)\n",
    "    col110 = fits.Column(name='ONC Total',format='F',unit='Counts',array=ONC)\n",
    "    col111 = fits.Column(name='OFD Total',format='F',unit='Counts',array=OFD)\n",
    "    col112 = fits.Column(name='OFC Total',format='F',unit='Counts',array=OFC)\n",
    "    col132 = fits.Column(name='YND Perc',format='F',unit='Percentage',array=YNDperc)\n",
    "    col133 = fits.Column(name='YNC Perc',format='F',unit='Percentage',array=YNCperc)\n",
    "    col134 = fits.Column(name='YFD Perc',format='F',unit='Percentage',array=YFDperc)\n",
    "    col135 = fits.Column(name='YFC Perc',format='F',unit='Percentage',array=YFCperc)\n",
    "    col136 = fits.Column(name='OND Perc',format='F',unit='Percentage',array=ONDperc)\n",
    "    col137 = fits.Column(name='ONC Perc',format='F',unit='Percentage',array=ONCperc)\n",
    "    col138 = fits.Column(name='OFD Perc',format='F',unit='Percentage',array=OFDperc)\n",
    "    col139 = fits.Column(name='OFC Perc',format='F',unit='Percentage',array=OFCperc)\n",
    "    col113 = fits.Column(name='Y Total',format='F',unit='Counts',array=Y)\n",
    "    col114 = fits.Column(name='O Total',format='F',unit='Counts',array=O)\n",
    "    col115 = fits.Column(name='N Total',format='F',unit='Counts',array=N)\n",
    "    col116 = fits.Column(name='F Total',format='F',unit='Counts',array=F)\n",
    "    col117 = fits.Column(name='D Total',format='F',unit='Counts',array=D)\n",
    "    col118 = fits.Column(name='C Total',format='F',unit='Counts',array=C)\n",
    "    col119 = fits.Column(name='ZY Total',format='F',unit='Counts',array=ZY)\n",
    "    col120 = fits.Column(name='ZJ Total',format='F',unit='Counts',array=ZJ)\n",
    "    col121 = fits.Column(name='ZH Total',format='F',unit='Counts',array=ZH)\n",
    "    col122 = fits.Column(name='ZK Total',format='F',unit='Counts',array=ZK)\n",
    "    col123 = fits.Column(name='YJ Total',format='F',unit='Counts',array=YJ)\n",
    "    col124 = fits.Column(name='YH Total',format='F',unit='Counts',array=YH)\n",
    "    col125 = fits.Column(name='YK Total',format='F',unit='Counts',array=YK)\n",
    "    col126 = fits.Column(name='JH Total',format='F',unit='Counts',array=JH)\n",
    "    col127 = fits.Column(name='JK Total',format='F',unit='Counts',array=JK)\n",
    "    col128 = fits.Column(name='HK Total',format='F',unit='Counts',array=HK)\n",
    "    col129 = fits.Column(name='Filters',format='F',unit='Number',array=Filters)\n",
    "    col130 = fits.Column(name='Total Counts',format='F',unit='Number',array=Total)\n",
    "    col131 = fits.Column(name='Percentage Count',format='F',unit='Percentage',array=Percentage)\n",
    "    col143 = fits.Column(name='Max_Percentage',format='F',unit='Percentage',array=np.array(Max_Perc))\n",
    "\n",
    "    cols = fits.ColDefs([col1,col2,col3,col4,col3a,col4a,col5,col6,col7,col8,col9,col10,\n",
    "                         col11,col12,col13,col14,col15,col16,col17,col18,col19,col20,\n",
    "                         col21,col22,col23,col24,col25,col26,col27,col28,col29,col30,\n",
    "                         col31,col32,col33,col34,col35,col36,col37,col38,col39,col40,\n",
    "                         col41,col42,col43,col44,col45,col46,col47,col48,col49,col50,\n",
    "                         col51,col52,col53,col54,col55,col56,col57,col58,col59,col60,\n",
    "                         col61,col62,col63,col64,col65,col66,col67,col68,col69,col70,\n",
    "                         col71,col72,col73,col74,col75,col76,col77,col78,col79,col80,\n",
    "                         col81,col82,col83,col84,col85,col86,col87,col88,col89,col90,\n",
    "                         col91,col92,col93,col94,col95,col96,col97,col98,col99,col100,\n",
    "                         col101,col102,col103,col104,col105,col132,col106,col133,col107,\n",
    "                         col134,col108,col135,col109,col136,col110,col137,col111,col138,\n",
    "                         col112,col139,col113,col114,col115,col116,col117,col118,col119,\n",
    "                         col120,col121,col122,col123,col124,col125,col126,col127,col128,\n",
    "                         col129,col130,col131,col143])\n",
    "    \n",
    "    '''Here we make the fits file'''\n",
    "\n",
    "    data = fits.BinTableHDU.from_columns(cols)\n",
    "    data.writeto(output_fits)\n",
    "\n",
    "    '''Now we make the reg file'''\n",
    "\n",
    "    pos_data = []\n",
    "\n",
    "    for v in range(len(RA_dms)):\n",
    "\n",
    "        pos_data.append([RA_dms[v],Dec_dms[v]])\n",
    "\n",
    "    with open(output_reg, 'w', newline='') as csvfile:\n",
    "        reg_file = csv.writer(csvfile, delimiter=',', quotechar=' ', quoting=csv.QUOTE_ALL)\n",
    "        for w in range(len(pos_data)):\n",
    "            reg_file.writerow([pos_data[w][0] , pos_data[w][1]])\n",
    "            \n",
    "def Distance(d=10):\n",
    "    '''This is just the distance modulus, for use in parsecs.'''\n",
    "    mu = 5*math.log(d,10)-5\n",
    "    return mu"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def vlm_object_added(file,model,output_name,percnum=.5):\n",
    "    \n",
    "    '''This is very similar to the previous code, but this one removes sources that do not have a high enough \n",
    "    percentage of flags (percnum), but adds back in any that have at least that percentage in an individual set-up.\n",
    "    In other words, this was used to determine the final candidate list.'''\n",
    "    \n",
    "    RA_dms = []\n",
    "    Dec_dms = []\n",
    "    RA_Rad = []\n",
    "    Dec_Rad = []\n",
    "    RA_ICRS = []\n",
    "    Dec_ICRS = []\n",
    "    zmag = []\n",
    "    ymag = []\n",
    "    jmag = []\n",
    "    hmag = []\n",
    "    kmag = []\n",
    "    zobsmag = []\n",
    "    yobsmag = []\n",
    "    jobsmag = []\n",
    "    hobsmag = []\n",
    "    kobsmag = []\n",
    "    zaper = []\n",
    "    yaper = []\n",
    "    japer = []\n",
    "    haper = []\n",
    "    kaper = []\n",
    "    zapererr = []\n",
    "    yapererr = []\n",
    "    japererr = []\n",
    "    hapererr = []\n",
    "    kapererr = []\n",
    "    \n",
    "    intest = [[[[[0 for z in range(5)] for y in range(4)] for x in range(2)] for w in range(2)] for v in range(2)]\n",
    "    \n",
    "    hdul = fits.open(file)\n",
    "    allpoints = list(range(len(hdul[1].data)))\n",
    "    \n",
    "    Filters = [0 for z in allpoints]\n",
    "    YND = []\n",
    "    YNC = []\n",
    "    YFD = []\n",
    "    YFC = []\n",
    "    OND = []\n",
    "    ONC = []\n",
    "    OFD = []\n",
    "    OFC = []\n",
    "    ZY = []\n",
    "    ZJ = []\n",
    "    ZH = []\n",
    "    ZK = []\n",
    "    YJ = []\n",
    "    YH = []\n",
    "    YK = []\n",
    "    JH = []\n",
    "    JK = []\n",
    "    HK = []\n",
    "    Max = [0 for z in allpoints]\n",
    "    Total = [0 for z in allpoints]\n",
    "    Percentage = [0 for z in allpoints]\n",
    "    Y = [0 for z in allpoints]\n",
    "    O = [0 for z in allpoints]\n",
    "    N = [0 for z in allpoints]\n",
    "    F = [0 for z in allpoints]\n",
    "    D = [0 for z in allpoints]\n",
    "    C = [0 for z in allpoints]\n",
    "    AgeDiff = [0 for z in allpoints]\n",
    "    DisDiff = [0 for z in allpoints]\n",
    "    ModDiff = [0 for z in allpoints]\n",
    "    YNDperc = [0 for z in allpoints]\n",
    "    YNCperc = [0 for z in allpoints]\n",
    "    YFDperc = [0 for z in allpoints]\n",
    "    YFCperc = [0 for z in allpoints]\n",
    "    ONDperc = [0 for z in allpoints]\n",
    "    ONCperc = [0 for z in allpoints]\n",
    "    OFDperc = [0 for z in allpoints]\n",
    "    OFCperc = [0 for z in allpoints]\n",
    "    Max_Perc = [0 for z in allpoints]\n",
    "\n",
    "    ra_dms = hdul[1].data['RA_dms']\n",
    "    dec_dms = hdul[1].data['Dec_dms']\n",
    "    ra_rad = hdul[1].data['RA_Rad']\n",
    "    dec_rad = hdul[1].data['Dec_Rad']\n",
    "    ra_icrs = hdul[1].data['RA_ICRS']\n",
    "    dec_icrs = hdul[1].data['Dec_ICRS']\n",
    "    mag_z = hdul[1].data['Mag_Z']\n",
    "    obs_mag_z = hdul[1].data['Obs_Mag_Z']\n",
    "    aper_z = hdul[1].data['Aper_Z']\n",
    "    aper_z_err = hdul[1].data['Aper_Z_err']\n",
    "    mag_y = hdul[1].data['Mag_Y']\n",
    "    obs_mag_y = hdul[1].data['Obs_Mag_Y']\n",
    "    aper_y = hdul[1].data['Aper_Y']\n",
    "    aper_y_err = hdul[1].data['Aper_Y_err']\n",
    "    mag_j = hdul[1].data['Mag_J']\n",
    "    obs_mag_j = hdul[1].data['Obs_Mag_J']\n",
    "    aper_j = hdul[1].data['Aper_J']\n",
    "    aper_j_err = hdul[1].data['Aper_J_err']\n",
    "    mag_h = hdul[1].data['Mag_H']\n",
    "    obs_mag_h = hdul[1].data['Obs_Mag_H']\n",
    "    aper_h = hdul[1].data['Aper_H']\n",
    "    aper_h_err = hdul[1].data['Aper_H_err']\n",
    "    mag_k = hdul[1].data['Mag_K']\n",
    "    obs_mag_k = hdul[1].data['Obs_Mag_K']\n",
    "    aper_k = hdul[1].data['Aper_K']\n",
    "    aper_k_err = hdul[1].data['Aper_K_err']\n",
    "    \n",
    "    err_z = aper_z_err/(aper_z)\n",
    "    err_y = aper_y_err/(aper_y)\n",
    "    err_j = aper_j_err/(aper_j)\n",
    "    err_h = aper_h_err/(aper_h)\n",
    "    err_k = aper_k_err/(aper_k)\n",
    "    \n",
    "    model = fits.open('model_iso.fits')\n",
    "\n",
    "    mass1 = model[1].data['Mass'][0:28]\n",
    "    teff1 = model[1].data['Teff'][0:28]\n",
    "    zmd1 = model[1].data['ZmagD'][0:28]\n",
    "    ymd1 = model[1].data['YmagD'][0:28]\n",
    "    jmd1 = model[1].data['JmagD'][0:28]\n",
    "    hmd1 = model[1].data['HmagD'][0:28]\n",
    "    kmd1 = model[1].data['KsmagD'][0:28]\n",
    "    zmc1 = model[1].data['ZmagC'][0:28]\n",
    "    ymc1 = model[1].data['YmagC'][0:28]\n",
    "    jmc1 = model[1].data['JmagC'][0:28]\n",
    "    hmc1 = model[1].data['HmagC'][0:28]\n",
    "    kmc1 = model[1].data['KsmagC'][0:28]\n",
    "    \n",
    "    mass5 = model[1].data['Mass'][28:57]\n",
    "    teff5 = model[1].data['Teff'][28:57]\n",
    "    zmd5 = model[1].data['ZmagD'][28:57]\n",
    "    ymd5 = model[1].data['YmagD'][28:57]\n",
    "    jmd5 = model[1].data['JmagD'][28:57]\n",
    "    hmd5 = model[1].data['HmagD'][28:57]\n",
    "    kmd5 = model[1].data['KsmagD'][28:57]\n",
    "    zmc5 = model[1].data['ZmagC'][28:57]\n",
    "    ymc5 = model[1].data['YmagC'][28:57]\n",
    "    jmc5 = model[1].data['JmagC'][28:57]\n",
    "    hmc5 = model[1].data['HmagC'][28:57]\n",
    "    kmc5 = model[1].data['KsmagC'][28:57]\n",
    "\n",
    "    massy_fit = np.polyfit(teff1, mass1, 14)\n",
    "    massy_poly = np.poly1d(massy_fit)\n",
    "    zcy_fit = np.polyfit(teff1, zmc1, 14)\n",
    "    zcy_poly = np.poly1d(zcy_fit)\n",
    "    zdy_fit = np.polyfit(teff1, zmd1, 14)\n",
    "    zdy_poly = np.poly1d(zdy_fit)\n",
    "    ycy_fit = np.polyfit(teff1, ymc1, 14)\n",
    "    ycy_poly = np.poly1d(ycy_fit)\n",
    "    ydy_fit = np.polyfit(teff1, ymd1, 14)\n",
    "    ydy_poly = np.poly1d(ydy_fit)\n",
    "    jcy_fit = np.polyfit(teff1, jmc1, 14)\n",
    "    jcy_poly = np.poly1d(jcy_fit)\n",
    "    jdy_fit = np.polyfit(teff1, jmd1, 14)\n",
    "    jdy_poly = np.poly1d(jdy_fit)\n",
    "    hcy_fit = np.polyfit(teff1, hmc1, 14)\n",
    "    hcy_poly = np.poly1d(hcy_fit)\n",
    "    hdy_fit = np.polyfit(teff1, hmd1, 14)\n",
    "    hdy_poly = np.poly1d(hdy_fit)\n",
    "    kcy_fit = np.polyfit(teff1, kmc1, 14)\n",
    "    kcy_poly = np.poly1d(kcy_fit)\n",
    "    kdy_fit = np.polyfit(teff1, kmd1, 14)\n",
    "    kdy_poly = np.poly1d(kdy_fit)\n",
    "    \n",
    "    masso_fit = np.polyfit(teff5, mass5, 14)\n",
    "    masso_poly = np.poly1d(masso_fit)\n",
    "    zco_fit = np.polyfit(teff5, zmc5, 14)\n",
    "    zco_poly = np.poly1d(zco_fit)\n",
    "    zdo_fit = np.polyfit(teff5, zmd5, 14)\n",
    "    zdo_poly = np.poly1d(zdo_fit)\n",
    "    yco_fit = np.polyfit(teff5, ymc5, 14)\n",
    "    yco_poly = np.poly1d(yco_fit)\n",
    "    ydo_fit = np.polyfit(teff5, ymd5, 14)\n",
    "    ydo_poly = np.poly1d(ydo_fit)\n",
    "    jco_fit = np.polyfit(teff5, jmc5, 14)\n",
    "    jco_poly = np.poly1d(jco_fit)\n",
    "    jdo_fit = np.polyfit(teff5, jmd5, 14)\n",
    "    jdo_poly = np.poly1d(jdo_fit)\n",
    "    hco_fit = np.polyfit(teff5, hmc5, 14)\n",
    "    hco_poly = np.poly1d(hco_fit)\n",
    "    hdo_fit = np.polyfit(teff5, hmd5, 14)\n",
    "    hdo_poly = np.poly1d(hdo_fit)\n",
    "    kco_fit = np.polyfit(teff5, kmc5, 14)\n",
    "    kco_poly = np.poly1d(kco_fit)\n",
    "    kdo_fit = np.polyfit(teff5, kmd5, 14)\n",
    "    kdo_poly = np.poly1d(kdo_fit)\n",
    "    \n",
    "    idz = np.isfinite(mag_z) & np.isfinite(err_z)\n",
    "    zerr_fit = np.polyfit(mag_z[idz],err_z[idz],8)\n",
    "    zerr_poly = np.poly1d(zerr_fit)\n",
    "    idy = np.isfinite(mag_y) & np.isfinite(err_y)\n",
    "    yerr_fit = np.polyfit(mag_y[idy],err_y[idy],8)\n",
    "    yerr_poly = np.poly1d(yerr_fit)\n",
    "    idj = np.isfinite(mag_j) & np.isfinite(err_j)\n",
    "    jerr_fit = np.polyfit(mag_j[idj],err_j[idj],8)\n",
    "    jerr_poly = np.poly1d(jerr_fit)\n",
    "    idh = np.isfinite(mag_h) & np.isfinite(err_h)\n",
    "    herr_fit = np.polyfit(mag_h[idh],err_h[idh],8)\n",
    "    herr_poly = np.poly1d(herr_fit)\n",
    "    idk = np.isfinite(mag_k) & np.isfinite(err_k)\n",
    "    kerr_fit = np.polyfit(mag_k[idk],err_k[idk],8)\n",
    "    kerr_poly = np.poly1d(kerr_fit)\n",
    "    \n",
    "    bd_range_1myr = range(1300,2931)\n",
    "    full_rangey = range(1300,4000)\n",
    "    full_rangeo = range(1300,4000)\n",
    "    full_rangec = range(1300,4000)\n",
    "    full_ranged = range(1650,4000)\n",
    "    \n",
    "    mu_far = Distance(420)\n",
    "    mu_near = Distance(350)\n",
    "    Av = 1.1\n",
    "    \n",
    "    Az = 0.5006359319 * Av\n",
    "    Ay = 0.3913227425 * Av\n",
    "    Aj = 0.281344478 * Av\n",
    "    Ah = 0.1812818241 * Av\n",
    "    Ak = 0.1180678823 * Av\n",
    "    \n",
    "    Azy = Az - Ay\n",
    "    Azj = Az - Aj\n",
    "    Azh = Az - Ah\n",
    "    Azk = Az - Ak\n",
    "    Ayj = Ay - Aj\n",
    "    Ayh = Ay - Ah\n",
    "    Ayk = Ay - Ak\n",
    "    Ajh = Aj - Ah\n",
    "    Ajk = Aj - Ak\n",
    "    Ahk = Ah - Ak\n",
    "    \n",
    "    poly = [[[zdy_poly(full_ranged),zdo_poly(full_ranged)],[zcy_poly(full_rangec),zco_poly(full_rangec)]],\n",
    "            [[ydy_poly(full_ranged),ydo_poly(full_ranged)],[ycy_poly(full_rangec),yco_poly(full_rangec)]],\n",
    "            [[jdy_poly(full_ranged),jdo_poly(full_ranged)],[jcy_poly(full_rangec),jco_poly(full_rangec)]],\n",
    "            [[hdy_poly(full_ranged),hdo_poly(full_ranged)],[hcy_poly(full_rangec),hco_poly(full_rangec)]],\n",
    "            [[kdy_poly(full_ranged),kdo_poly(full_ranged)],[kcy_poly(full_rangec),kco_poly(full_rangec)]]]\n",
    "    \n",
    "    observed_data = [mag_z,mag_y,mag_j,mag_h,mag_k]\n",
    "    observed_array = np.array(observed_data)\n",
    "\n",
    "    ext = [[Az,Ay,Aj,Ah,Ak],[[0,Azy,Azj,Azh,Azk],[0,0,Ayj,Ayh,Ayk],[0,0,0,Ajh,Ajk],[0,0,0,0,Ahk]]]\n",
    "    \n",
    "    mu = [mu_near,mu_far]\n",
    "    \n",
    "    for alpha in range(2):\n",
    "        for beta in range(2):\n",
    "            for gamma in range(2):\n",
    "                for delta in range(4):\n",
    "                    for epsilon in range(delta+1,5):\n",
    "                        \n",
    "                        err_poly_max = [zerr_poly(poly[delta][gamma][beta] + mu[alpha] + 1*ext[0][delta]),\n",
    "                                    yerr_poly(poly[delta][gamma][beta] + mu[alpha] + 1*ext[0][delta]),\n",
    "                                    jerr_poly(poly[delta][gamma][beta] + mu[alpha] + 1*ext[0][delta]),\n",
    "                                    herr_poly(poly[delta][gamma][beta] + mu[alpha] + 1*ext[0][delta]),\n",
    "                                    kerr_poly(poly[delta][gamma][beta] + mu[alpha] + 1*ext[0][delta])]\n",
    "                        \n",
    "                        err_poly_max_ep = [zerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 1*ext[0][epsilon]),\n",
    "                                    yerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 1*ext[0][epsilon]),\n",
    "                                    jerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 1*ext[0][epsilon]),\n",
    "                                    herr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 1*ext[0][epsilon]),\n",
    "                                    kerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 1*ext[0][epsilon])]\n",
    "                        \n",
    "                        err_poly_no = [zerr_poly(poly[delta][gamma][beta] + mu[alpha] + 0*ext[0][delta]),\n",
    "                                    yerr_poly(poly[delta][gamma][beta] + mu[alpha] + 0*ext[0][delta]),\n",
    "                                    jerr_poly(poly[delta][gamma][beta] + mu[alpha] + 0*ext[0][delta]),\n",
    "                                    herr_poly(poly[delta][gamma][beta] + mu[alpha] + 0*ext[0][delta]),\n",
    "                                    kerr_poly(poly[delta][gamma][beta] + mu[alpha] + 0*ext[0][delta])]\n",
    "                        \n",
    "                        err_poly_no_ep = [zerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 0*ext[0][epsilon]),\n",
    "                                    yerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 0*ext[0][epsilon]),\n",
    "                                    jerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 0*ext[0][epsilon]),\n",
    "                                    herr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 0*ext[0][epsilon]),\n",
    "                                    kerr_poly(poly[epsilon][gamma][beta] + mu[alpha] + 0*ext[0][epsilon])]\n",
    "                            \n",
    "                        maxext = list(zip((poly[delta][gamma][beta] - \n",
    "                                           poly[epsilon][gamma][beta] + \n",
    "                                           (1*ext[1][delta][epsilon])) + \n",
    "                                          np.sqrt(err_poly_max[delta]**2 + err_poly_max_ep[epsilon]**2),\n",
    "                                          poly[delta][gamma][beta] + mu[alpha] + \n",
    "                                          (1*ext[0][delta]) - err_poly_max[delta]))\n",
    "                        noext = list(zip((poly[delta][gamma][beta] - \n",
    "                                          poly[epsilon][gamma][beta] + \n",
    "                                          (0*ext[1][delta][epsilon])) - \n",
    "                                         np.sqrt(err_poly_no[delta]**2 + err_poly_no_ep[epsilon]**2),\n",
    "                                         poly[delta][gamma][beta] + mu[alpha] + \n",
    "                                         (0*ext[0][delta]) + err_poly_no[delta]))\n",
    "                        \n",
    "                        sidepoints = maxext + noext[::-1]\n",
    "                        polypath = path.Path(sidepoints, closed=True)\n",
    "                        \n",
    "                        points = list(zip(observed_data[delta]-observed_data[epsilon],observed_data[delta]))\n",
    "                        intest[beta][alpha][gamma][delta][epsilon] = polypath.contains_points(points)\n",
    "    \n",
    "    for i in allpoints:\n",
    "        Filters[i] = np.sum(~np.isnan(observed_array[:,i]))\n",
    "        \n",
    "        outynd = []\n",
    "        outync = []\n",
    "        outyfd = []\n",
    "        outyfc = []\n",
    "        outond = []\n",
    "        outonc = []\n",
    "        outofd = []\n",
    "        outofc = []\n",
    "\n",
    "        for d in range(4):\n",
    "            for e in range(d+1,5):\n",
    "                outynd.append(intest[0][0][0][d][e][i])\n",
    "                outync.append(intest[0][0][1][d][e][i])\n",
    "                outyfd.append(intest[0][1][0][d][e][i])\n",
    "                outyfc.append(intest[0][1][1][d][e][i])\n",
    "                outond.append(intest[1][0][0][d][e][i])\n",
    "                outonc.append(intest[1][0][1][d][e][i])\n",
    "                outofd.append(intest[1][1][0][d][e][i])\n",
    "                outofc.append(intest[1][1][1][d][e][i])\n",
    "        YND.append(np.sum(outynd))\n",
    "        YNC.append(np.sum(outync))\n",
    "        YFD.append(np.sum(outyfd))\n",
    "        YFC.append(np.sum(outyfc))\n",
    "        OND.append(np.sum(outond))\n",
    "        ONC.append(np.sum(outonc))\n",
    "        OFD.append(np.sum(outofd))\n",
    "        OFC.append(np.sum(outofc))\n",
    "\n",
    "        outzy = []\n",
    "        outzj = []\n",
    "        outzh = []\n",
    "        outzk = []\n",
    "        outyj = []\n",
    "        outyh = []\n",
    "        outyk = []\n",
    "        outjh = []\n",
    "        outjk = []\n",
    "        outhk = []\n",
    "\n",
    "        for a in range(2):\n",
    "            for b in range(2):\n",
    "                for c in range(2):\n",
    "                    outzy.append(intest[a][b][c][0][1][i])\n",
    "                    outzj.append(intest[a][b][c][0][2][i])\n",
    "                    outzh.append(intest[a][b][c][0][3][i])\n",
    "                    outzk.append(intest[a][b][c][0][4][i])\n",
    "                    outyj.append(intest[a][b][c][1][2][i])\n",
    "                    outyh.append(intest[a][b][c][1][3][i])\n",
    "                    outyk.append(intest[a][b][c][1][4][i])\n",
    "                    outjh.append(intest[a][b][c][2][3][i])\n",
    "                    outjk.append(intest[a][b][c][2][4][i])\n",
    "                    outhk.append(intest[a][b][c][3][4][i])\n",
    "        ZY.append(np.sum(outzy))\n",
    "        ZJ.append(np.sum(outzj))\n",
    "        ZH.append(np.sum(outzh))\n",
    "        ZK.append(np.sum(outzk))\n",
    "        YJ.append(np.sum(outyj))\n",
    "        YH.append(np.sum(outyh))\n",
    "        YK.append(np.sum(outyk))\n",
    "        JH.append(np.sum(outjh))\n",
    "        JK.append(np.sum(outjk))\n",
    "        HK.append(np.sum(outhk))\n",
    "        \n",
    "        if Filters[i] == 2:\n",
    "            Max[i] = 1*8\n",
    "        if Filters[i] == 3:\n",
    "            Max[i] = 3*8\n",
    "        if Filters[i] == 4:\n",
    "            Max[i] = 6*8\n",
    "        if Filters[i] == 5:\n",
    "            Max[i] = 10*8\n",
    "           \n",
    "        Y[i] = np.sum(YND[i] + YNC[i] + YFD[i] + YFC[i])\n",
    "        O[i] = np.sum(OND[i] + ONC[i] + OFD[i] + OFC[i])\n",
    "        N[i] = np.sum(YND[i] + YNC[i] + OND[i] + ONC[i])\n",
    "        F[i] = np.sum(YFD[i] + YFC[i] + OFD[i] + OFC[i])\n",
    "        D[i] = np.sum(YND[i] + OND[i] + YFD[i] + OFD[i])\n",
    "        C[i] = np.sum(YNC[i] + ONC[i] + YFC[i] + OFC[i])\n",
    "        AgeDiff[i] = Y[i] - O[i]\n",
    "        DisDiff[i] = N[i] - F[i]\n",
    "        ModDiff[i] = D[i] - C[i]\n",
    "    \n",
    "        Total[i] = sum([YND[i],YNC[i],YFD[i],YFC[i],OND[i],ONC[i],OFD[i],OFC[i]])\n",
    "        YNDperc[i] = YND[i]/(Max[i]/8)\n",
    "        YNCperc[i] = YNC[i]/(Max[i]/8)\n",
    "        YFDperc[i] = YFD[i]/(Max[i]/8)\n",
    "        YFCperc[i] = YFC[i]/(Max[i]/8)\n",
    "        ONDperc[i] = OND[i]/(Max[i]/8)\n",
    "        ONCperc[i] = ONC[i]/(Max[i]/8)\n",
    "        OFDperc[i] = OFD[i]/(Max[i]/8)\n",
    "        OFCperc[i] = OFC[i]/(Max[i]/8)\n",
    "        Percentage[i] = Total[i]/Max[i]\n",
    "        Max_Perc[i] = max(YNDperc[i],YNCperc[i],YFDperc[i],YFCperc[i],\n",
    "                          ONDperc[i],ONCperc[i],OFDperc[i],OFCperc[i],Percentage[i])\n",
    "    \n",
    "    YND_zy = intest[0][0][0][0][1]\n",
    "    YND_zj = intest[0][0][0][0][2]\n",
    "    YND_zh = intest[0][0][0][0][3]\n",
    "    YND_zk = intest[0][0][0][0][4]\n",
    "    YND_yj = intest[0][0][0][1][2]\n",
    "    YND_yh = intest[0][0][0][1][3]\n",
    "    YND_yk = intest[0][0][0][1][4]\n",
    "    YND_jh = intest[0][0][0][2][3]\n",
    "    YND_jk = intest[0][0][0][2][4]\n",
    "    YND_hk = intest[0][0][0][3][4]\n",
    "    YNC_zy = intest[0][0][1][0][1]\n",
    "    YNC_zj = intest[0][0][1][0][2]\n",
    "    YNC_zh = intest[0][0][1][0][3]\n",
    "    YNC_zk = intest[0][0][1][0][4]\n",
    "    YNC_yj = intest[0][0][1][1][2]\n",
    "    YNC_yh = intest[0][0][1][1][3]\n",
    "    YNC_yk = intest[0][0][1][1][4]\n",
    "    YNC_jh = intest[0][0][1][2][3]\n",
    "    YNC_jk = intest[0][0][1][2][4]\n",
    "    YNC_hk = intest[0][0][1][3][4]\n",
    "    YFD_zy = intest[0][1][0][0][1]\n",
    "    YFD_zj = intest[0][1][0][0][2]\n",
    "    YFD_zh = intest[0][1][0][0][3]\n",
    "    YFD_zk = intest[0][1][0][0][4]\n",
    "    YFD_yj = intest[0][1][0][1][2]\n",
    "    YFD_yh = intest[0][1][0][1][3]\n",
    "    YFD_yk = intest[0][1][0][1][4]\n",
    "    YFD_jh = intest[0][1][0][2][3]\n",
    "    YFD_jk = intest[0][1][0][2][4]\n",
    "    YFD_hk = intest[0][1][0][3][4]\n",
    "    YFC_zy = intest[0][1][1][0][1]\n",
    "    YFC_zj = intest[0][1][1][0][2]\n",
    "    YFC_zh = intest[0][1][1][0][3]\n",
    "    YFC_zk = intest[0][1][1][0][4]\n",
    "    YFC_yj = intest[0][1][1][1][2]\n",
    "    YFC_yh = intest[0][1][1][1][3]\n",
    "    YFC_yk = intest[0][1][1][1][4]\n",
    "    YFC_jh = intest[0][1][1][2][3]\n",
    "    YFC_jk = intest[0][1][1][2][4]\n",
    "    YFC_hk = intest[0][1][1][3][4]\n",
    "    OND_zy = intest[1][0][0][0][1]\n",
    "    OND_zj = intest[1][0][0][0][2]\n",
    "    OND_zh = intest[1][0][0][0][3]\n",
    "    OND_zk = intest[1][0][0][0][4]\n",
    "    OND_yj = intest[1][0][0][1][2]\n",
    "    OND_yh = intest[1][0][0][1][3]\n",
    "    OND_yk = intest[1][0][0][1][4]\n",
    "    OND_jh = intest[1][0][0][2][3]\n",
    "    OND_jk = intest[1][0][0][2][4]\n",
    "    OND_hk = intest[1][0][0][3][4]\n",
    "    ONC_zy = intest[1][0][1][0][1]\n",
    "    ONC_zj = intest[1][0][1][0][2]\n",
    "    ONC_zh = intest[1][0][1][0][3]\n",
    "    ONC_zk = intest[1][0][1][0][4]\n",
    "    ONC_yj = intest[1][0][1][1][2]\n",
    "    ONC_yh = intest[1][0][1][1][3]\n",
    "    ONC_yk = intest[1][0][1][1][4]\n",
    "    ONC_jh = intest[1][0][1][2][3]\n",
    "    ONC_jk = intest[1][0][1][2][4]\n",
    "    ONC_hk = intest[1][0][1][3][4]\n",
    "    OFD_zy = intest[1][1][0][0][1]\n",
    "    OFD_zj = intest[1][1][0][0][2]\n",
    "    OFD_zh = intest[1][1][0][0][3]\n",
    "    OFD_zk = intest[1][1][0][0][4]\n",
    "    OFD_yj = intest[1][1][0][1][2]\n",
    "    OFD_yh = intest[1][1][0][1][3]\n",
    "    OFD_yk = intest[1][1][0][1][4]\n",
    "    OFD_jh = intest[1][1][0][2][3]\n",
    "    OFD_jk = intest[1][1][0][2][4]\n",
    "    OFD_hk = intest[1][1][0][3][4]\n",
    "    OFC_zy = intest[1][1][1][0][1]\n",
    "    OFC_zj = intest[1][1][1][0][2]\n",
    "    OFC_zh = intest[1][1][1][0][3]\n",
    "    OFC_zk = intest[1][1][1][0][4]\n",
    "    OFC_yj = intest[1][1][1][1][2]\n",
    "    OFC_yh = intest[1][1][1][1][3]\n",
    "    OFC_yk = intest[1][1][1][1][4]\n",
    "    OFC_jh = intest[1][1][1][2][3]\n",
    "    OFC_jk = intest[1][1][1][2][4]\n",
    "    OFC_hk = intest[1][1][1][3][4]\n",
    "    \n",
    "    output_fits = output_name + '.fits'\n",
    "    output_reg = output_name + '.reg'\n",
    "    \n",
    "    chosen = []\n",
    "    \n",
    "    for j in allpoints:\n",
    "        RA_dms.append(hdul[1].data['RA_dms'][j])\n",
    "        Dec_dms.append(hdul[1].data['Dec_dms'][j])\n",
    "        RA_Rad.append(hdul[1].data['RA_Rad'][j])\n",
    "        Dec_Rad.append(hdul[1].data['Dec_Rad'][j])\n",
    "        RA_ICRS.append(hdul[1].data['RA_ICRS'][j])\n",
    "        Dec_ICRS.append(hdul[1].data['Dec_ICRS'][j])\n",
    "        zmag.append(hdul[1].data['Mag_Z'][j])\n",
    "        ymag.append(hdul[1].data['Mag_Y'][j])\n",
    "        jmag.append(hdul[1].data['Mag_J'][j])\n",
    "        hmag.append(hdul[1].data['Mag_H'][j])\n",
    "        kmag.append(hdul[1].data['Mag_K'][j])\n",
    "        zobsmag.append(hdul[1].data['Obs_Mag_Z'][j])\n",
    "        yobsmag.append(hdul[1].data['Obs_Mag_Y'][j])\n",
    "        jobsmag.append(hdul[1].data['Obs_Mag_J'][j])\n",
    "        hobsmag.append(hdul[1].data['Obs_Mag_H'][j])\n",
    "        kobsmag.append(hdul[1].data['Obs_Mag_K'][j])\n",
    "        zaper.append(hdul[1].data['Aper_Z'][j])\n",
    "        yaper.append(hdul[1].data['Aper_Y'][j])\n",
    "        japer.append(hdul[1].data['Aper_J'][j])\n",
    "        haper.append(hdul[1].data['Aper_H'][j])\n",
    "        kaper.append(hdul[1].data['Aper_K'][j])\n",
    "        zapererr.append(hdul[1].data['Aper_Z_err'][j])\n",
    "        yapererr.append(hdul[1].data['Aper_Y_err'][j])\n",
    "        japererr.append(hdul[1].data['Aper_J_err'][j])\n",
    "        hapererr.append(hdul[1].data['Aper_H_err'][j])\n",
    "        kapererr.append(hdul[1].data['Aper_K_err'][j])\n",
    "        \n",
    "        if Percentage[j] >= percnum:\n",
    "            chosen.append(j)\n",
    "        else:\n",
    "            if ZJ[j] > 3:\n",
    "                if YNDperc[j] >= percnum:\n",
    "                    chosen.append(j)\n",
    "                elif YNCperc[j] >= percnum:\n",
    "                    chosen.append(j)\n",
    "                elif YFDperc[j] >= percnum:\n",
    "                    chosen.append(j)\n",
    "                elif YFCperc[j] >= percnum:\n",
    "                    chosen.append(j)\n",
    "                elif ONDperc[j] >= percnum:\n",
    "                    chosen.append(j)\n",
    "                elif ONCperc[j] >= percnum:\n",
    "                    chosen.append(j)\n",
    "                elif OFDperc[j] >= percnum:\n",
    "                    chosen.append(j)\n",
    "                elif OFCperc[j] >= percnum:\n",
    "                    chosen.append(j)\n",
    "    \n",
    "    col1 = fits.Column(name='RA_dms',format='11A',unit='dms',array=np.array(RA_dms)[chosen])\n",
    "    col2 = fits.Column(name='Dec_dms',format='11A',unit='dms',array=np.array(Dec_dms)[chosen])\n",
    "    col3 = fits.Column(name='RA_Rad',format='D',unit='radians',array=np.array(RA_Rad)[chosen])\n",
    "    col4 = fits.Column(name='Dec_Rad',format='D',unit='radians',array=np.array(Dec_Rad)[chosen])\n",
    "    col3a = fits.Column(name='RA_ICRS',format='D',unit='deg',array=np.array(RA_ICRS)[chosen])\n",
    "    col4a = fits.Column(name='Dec_ICRS',format='D',unit='deg',array=np.array(Dec_ICRS)[chosen])\n",
    "    col5 = fits.Column(name='Mag_Z',format='D',unit='mag',array=np.array(zmag)[chosen])\n",
    "    col6 = fits.Column(name='Obs_Mag_Z',format='D',unit='mag',array=np.array(zobsmag)[chosen])\n",
    "    col7 = fits.Column(name='Aper_Z',format='E',unit='ADU',array=np.array(zaper)[chosen])\n",
    "    col8 = fits.Column(name='Aper_Z_err',format='E',unit='ADU',array=np.array(zapererr)[chosen])\n",
    "    col9 = fits.Column(name='Mag_Y',format='D',unit='mag',array=np.array(ymag)[chosen])\n",
    "    col10 = fits.Column(name='Obs_Mag_Y',format='D',unit='mag',array=np.array(yobsmag)[chosen])\n",
    "    col11 = fits.Column(name='Aper_Y',format='E',unit='ADU',array=np.array(yaper)[chosen])\n",
    "    col12 = fits.Column(name='Aper_Y_err',format='E',unit='ADU',array=np.array(yapererr)[chosen])\n",
    "    col13 = fits.Column(name='Mag_J',format='D',unit='mag',array=np.array(jmag)[chosen])\n",
    "    col14 = fits.Column(name='Obs_Mag_J',format='D',unit='mag',array=np.array(jobsmag)[chosen])\n",
    "    col15 = fits.Column(name='Aper_J',format='E',unit='ADU',array=np.array(japer)[chosen])\n",
    "    col16 = fits.Column(name='Aper_J_err',format='E',unit='ADU',array=np.array(japererr)[chosen])\n",
    "    col17 = fits.Column(name='Mag_H',format='D',unit='mag',array=np.array(hmag)[chosen])\n",
    "    col18 = fits.Column(name='Obs_Mag_H',format='D',unit='mag',array=np.array(hobsmag)[chosen])\n",
    "    col19 = fits.Column(name='Aper_H',format='E',unit='ADU',array=np.array(haper)[chosen])\n",
    "    col20 = fits.Column(name='Aper_H_err',format='E',unit='ADU',array=np.array(hapererr)[chosen])\n",
    "    col21 = fits.Column(name='Mag_K',format='D',unit='mag',array=np.array(kmag)[chosen])\n",
    "    col22 = fits.Column(name='Obs_Mag_K',format='D',unit='mag',array=np.array(kobsmag)[chosen])\n",
    "    col23 = fits.Column(name='Aper_K',format='E',unit='ADU',array=np.array(kaper)[chosen])\n",
    "    col24 = fits.Column(name='Aper_K_err',format='E',unit='ADU',array=np.array(kapererr)[chosen])\n",
    "    col25 = fits.Column(name='YND_zy',format='11A',unit='Flag',array=np.array(YND_zy)[chosen])\n",
    "    col26 = fits.Column(name='YND_zj',format='11A',unit='Flag',array=np.array(YND_zj)[chosen])\n",
    "    col27 = fits.Column(name='YND_zh',format='11A',unit='Flag',array=np.array(YND_zh)[chosen])\n",
    "    col28 = fits.Column(name='YND_zk',format='11A',unit='Flag',array=np.array(YND_zk)[chosen])\n",
    "    col29 = fits.Column(name='YND_yj',format='11A',unit='Flag',array=np.array(YND_yj)[chosen])\n",
    "    col30 = fits.Column(name='YND_yh',format='11A',unit='Flag',array=np.array(YND_yh)[chosen])\n",
    "    col31 = fits.Column(name='YND_yk',format='11A',unit='Flag',array=np.array(YND_yk)[chosen])\n",
    "    col32 = fits.Column(name='YND_jh',format='11A',unit='Flag',array=np.array(YND_jh)[chosen])\n",
    "    col33 = fits.Column(name='YND_jk',format='11A',unit='Flag',array=np.array(YND_jk)[chosen])\n",
    "    col34 = fits.Column(name='YND_hk',format='11A',unit='Flag',array=np.array(YND_hk)[chosen])\n",
    "    col35 = fits.Column(name='YNC_zy',format='11A',unit='Flag',array=np.array(YNC_zy)[chosen])\n",
    "    col36 = fits.Column(name='YNC_zj',format='11A',unit='Flag',array=np.array(YNC_zj)[chosen])\n",
    "    col37 = fits.Column(name='YNC_zh',format='11A',unit='Flag',array=np.array(YNC_zh)[chosen])\n",
    "    col38 = fits.Column(name='YNC_zk',format='11A',unit='Flag',array=np.array(YNC_zk)[chosen])\n",
    "    col39 = fits.Column(name='YNC_yj',format='11A',unit='Flag',array=np.array(YNC_yj)[chosen])\n",
    "    col40 = fits.Column(name='YNC_yh',format='11A',unit='Flag',array=np.array(YNC_yh)[chosen])\n",
    "    col41 = fits.Column(name='YNC_yk',format='11A',unit='Flag',array=np.array(YNC_yk)[chosen])\n",
    "    col42 = fits.Column(name='YNC_jh',format='11A',unit='Flag',array=np.array(YNC_jh)[chosen])\n",
    "    col43 = fits.Column(name='YNC_jk',format='11A',unit='Flag',array=np.array(YNC_jk)[chosen])\n",
    "    col44 = fits.Column(name='YNC_hk',format='11A',unit='Flag',array=np.array(YNC_hk)[chosen])\n",
    "    col45 = fits.Column(name='YFD_zy',format='11A',unit='Flag',array=np.array(YFD_zy)[chosen])\n",
    "    col46 = fits.Column(name='YFD_zj',format='11A',unit='Flag',array=np.array(YFD_zj)[chosen])\n",
    "    col47 = fits.Column(name='YFD_zh',format='11A',unit='Flag',array=np.array(YFD_zh)[chosen])\n",
    "    col48 = fits.Column(name='YFD_zk',format='11A',unit='Flag',array=np.array(YFD_zk)[chosen])\n",
    "    col49 = fits.Column(name='YFD_yj',format='11A',unit='Flag',array=np.array(YFD_yj)[chosen])\n",
    "    col50 = fits.Column(name='YFD_yh',format='11A',unit='Flag',array=np.array(YFD_yh)[chosen])\n",
    "    col51 = fits.Column(name='YFD_yk',format='11A',unit='Flag',array=np.array(YFD_yk)[chosen])\n",
    "    col52 = fits.Column(name='YFD_jh',format='11A',unit='Flag',array=np.array(YFD_jh)[chosen])\n",
    "    col53 = fits.Column(name='YFD_jk',format='11A',unit='Flag',array=np.array(YFD_jk)[chosen])\n",
    "    col54 = fits.Column(name='YFD_hk',format='11A',unit='Flag',array=np.array(YFD_hk)[chosen])\n",
    "    col55 = fits.Column(name='YFC_zy',format='11A',unit='Flag',array=np.array(YFC_zy)[chosen])\n",
    "    col56 = fits.Column(name='YFC_zj',format='11A',unit='Flag',array=np.array(YFC_zj)[chosen])\n",
    "    col57 = fits.Column(name='YFC_zh',format='11A',unit='Flag',array=np.array(YFC_zh)[chosen])\n",
    "    col58 = fits.Column(name='YFC_zk',format='11A',unit='Flag',array=np.array(YFC_zk)[chosen])\n",
    "    col59 = fits.Column(name='YFC_yj',format='11A',unit='Flag',array=np.array(YFC_yj)[chosen])\n",
    "    col60 = fits.Column(name='YFC_yh',format='11A',unit='Flag',array=np.array(YFC_yh)[chosen])\n",
    "    col61 = fits.Column(name='YFC_yk',format='11A',unit='Flag',array=np.array(YFC_yk)[chosen])\n",
    "    col62 = fits.Column(name='YFC_jh',format='11A',unit='Flag',array=np.array(YFC_jh)[chosen])\n",
    "    col63 = fits.Column(name='YFC_jk',format='11A',unit='Flag',array=np.array(YFC_jk)[chosen])\n",
    "    col64 = fits.Column(name='YFC_hk',format='11A',unit='Flag',array=np.array(YFC_hk)[chosen])\n",
    "    col65 = fits.Column(name='OND_zy',format='11A',unit='Flag',array=np.array(OND_zy)[chosen])\n",
    "    col66 = fits.Column(name='OND_zj',format='11A',unit='Flag',array=np.array(OND_zj)[chosen])\n",
    "    col67 = fits.Column(name='OND_zh',format='11A',unit='Flag',array=np.array(OND_zh)[chosen])\n",
    "    col68 = fits.Column(name='OND_zk',format='11A',unit='Flag',array=np.array(OND_zk)[chosen])\n",
    "    col69 = fits.Column(name='OND_yj',format='11A',unit='Flag',array=np.array(OND_yj)[chosen])\n",
    "    col70 = fits.Column(name='OND_yh',format='11A',unit='Flag',array=np.array(OND_yh)[chosen])\n",
    "    col71 = fits.Column(name='OND_yk',format='11A',unit='Flag',array=np.array(OND_yk)[chosen])\n",
    "    col72 = fits.Column(name='OND_jh',format='11A',unit='Flag',array=np.array(OND_jh)[chosen])\n",
    "    col73 = fits.Column(name='OND_jk',format='11A',unit='Flag',array=np.array(OND_jk)[chosen])\n",
    "    col74 = fits.Column(name='OND_hk',format='11A',unit='Flag',array=np.array(OND_hk)[chosen])\n",
    "    col75 = fits.Column(name='ONC_zy',format='11A',unit='Flag',array=np.array(ONC_zy)[chosen])\n",
    "    col76 = fits.Column(name='ONC_zj',format='11A',unit='Flag',array=np.array(ONC_zj)[chosen])\n",
    "    col77 = fits.Column(name='ONC_zh',format='11A',unit='Flag',array=np.array(ONC_zh)[chosen])\n",
    "    col78 = fits.Column(name='ONC_zk',format='11A',unit='Flag',array=np.array(ONC_zk)[chosen])\n",
    "    col79 = fits.Column(name='ONC_yj',format='11A',unit='Flag',array=np.array(ONC_yj)[chosen])\n",
    "    col80 = fits.Column(name='ONC_yh',format='11A',unit='Flag',array=np.array(ONC_yh)[chosen])\n",
    "    col81 = fits.Column(name='ONC_yk',format='11A',unit='Flag',array=np.array(ONC_yk)[chosen])\n",
    "    col82 = fits.Column(name='ONC_jh',format='11A',unit='Flag',array=np.array(ONC_jh)[chosen])\n",
    "    col83 = fits.Column(name='ONC_jk',format='11A',unit='Flag',array=np.array(ONC_jk)[chosen])\n",
    "    col84 = fits.Column(name='ONC_hk',format='11A',unit='Flag',array=np.array(ONC_hk)[chosen])\n",
    "    col85 = fits.Column(name='OFD_zy',format='11A',unit='Flag',array=np.array(OFD_zy)[chosen])\n",
    "    col86 = fits.Column(name='OFD_zj',format='11A',unit='Flag',array=np.array(OFD_zj)[chosen])\n",
    "    col87 = fits.Column(name='OFD_zh',format='11A',unit='Flag',array=np.array(OFD_zh)[chosen])\n",
    "    col88 = fits.Column(name='OFD_zk',format='11A',unit='Flag',array=np.array(OFD_zk)[chosen])\n",
    "    col89 = fits.Column(name='OFD_yj',format='11A',unit='Flag',array=np.array(OFD_yj)[chosen])\n",
    "    col90 = fits.Column(name='OFD_yh',format='11A',unit='Flag',array=np.array(OFD_yh)[chosen])\n",
    "    col91 = fits.Column(name='OFD_yk',format='11A',unit='Flag',array=np.array(OFD_yk)[chosen])\n",
    "    col92 = fits.Column(name='OFD_jh',format='11A',unit='Flag',array=np.array(OFD_jh)[chosen])\n",
    "    col93 = fits.Column(name='OFD_jk',format='11A',unit='Flag',array=np.array(OFD_jk)[chosen])\n",
    "    col94 = fits.Column(name='OFD_hk',format='11A',unit='Flag',array=np.array(OFD_hk)[chosen])\n",
    "    col95 = fits.Column(name='OFC_zy',format='11A',unit='Flag',array=np.array(OFC_zy)[chosen])\n",
    "    col96 = fits.Column(name='OFC_zj',format='11A',unit='Flag',array=np.array(OFC_zj)[chosen])\n",
    "    col97 = fits.Column(name='OFC_zh',format='11A',unit='Flag',array=np.array(OFC_zh)[chosen])\n",
    "    col98 = fits.Column(name='OFC_zk',format='11A',unit='Flag',array=np.array(OFC_zk)[chosen])\n",
    "    col99 = fits.Column(name='OFC_yj',format='11A',unit='Flag',array=np.array(OFC_yj)[chosen])\n",
    "    col100 = fits.Column(name='OFC_yh',format='11A',unit='Flag',array=np.array(OFC_yh)[chosen])\n",
    "    col101 = fits.Column(name='OFC_yk',format='11A',unit='Flag',array=np.array(OFC_yk)[chosen])\n",
    "    col102 = fits.Column(name='OFC_jh',format='11A',unit='Flag',array=np.array(OFC_jh)[chosen])\n",
    "    col103 = fits.Column(name='OFC_jk',format='11A',unit='Flag',array=np.array(OFC_jk)[chosen])\n",
    "    col104 = fits.Column(name='OFC_hk',format='11A',unit='Flag',array=np.array(OFC_hk)[chosen])\n",
    "    col105 = fits.Column(name='YND Total',format='F',unit='Counts',array=np.array(YND)[chosen])\n",
    "    col106 = fits.Column(name='YNC Total',format='F',unit='Counts',array=np.array(YNC)[chosen])\n",
    "    col107 = fits.Column(name='YFD Total',format='F',unit='Counts',array=np.array(YFD)[chosen])\n",
    "    col108 = fits.Column(name='YFC Total',format='F',unit='Counts',array=np.array(YFC)[chosen])\n",
    "    col109 = fits.Column(name='OND Total',format='F',unit='Counts',array=np.array(OND)[chosen])\n",
    "    col110 = fits.Column(name='ONC Total',format='F',unit='Counts',array=np.array(ONC)[chosen])\n",
    "    col111 = fits.Column(name='OFD Total',format='F',unit='Counts',array=np.array(OFD)[chosen])\n",
    "    col112 = fits.Column(name='OFC Total',format='F',unit='Counts',array=np.array(OFC)[chosen])\n",
    "    col132 = fits.Column(name='YND Perc',format='F',unit='Percentage',array=np.array(YNDperc)[chosen])\n",
    "    col133 = fits.Column(name='YNC Perc',format='F',unit='Percentage',array=np.array(YNCperc)[chosen])\n",
    "    col134 = fits.Column(name='YFD Perc',format='F',unit='Percentage',array=np.array(YFDperc)[chosen])\n",
    "    col135 = fits.Column(name='YFC Perc',format='F',unit='Percentage',array=np.array(YFCperc)[chosen])\n",
    "    col136 = fits.Column(name='OND Perc',format='F',unit='Percentage',array=np.array(ONDperc)[chosen])\n",
    "    col137 = fits.Column(name='ONC Perc',format='F',unit='Percentage',array=np.array(ONCperc)[chosen])\n",
    "    col138 = fits.Column(name='OFD Perc',format='F',unit='Percentage',array=np.array(OFDperc)[chosen])\n",
    "    col139 = fits.Column(name='OFC Perc',format='F',unit='Percentage',array=np.array(OFCperc)[chosen])\n",
    "    col113 = fits.Column(name='Y Total',format='F',unit='Counts',array=np.array(Y)[chosen])\n",
    "    col114 = fits.Column(name='O Total',format='F',unit='Counts',array=np.array(O)[chosen])\n",
    "    col140 = fits.Column(name='Age Diff',format='F',unit='Counts',array=np.array(AgeDiff)[chosen])\n",
    "    col115 = fits.Column(name='N Total',format='F',unit='Counts',array=np.array(N)[chosen])\n",
    "    col116 = fits.Column(name='F Total',format='F',unit='Counts',array=np.array(F)[chosen])\n",
    "    col141 = fits.Column(name='Distance Diff',format='F',unit='Counts',array=np.array(DisDiff)[chosen])\n",
    "    col117 = fits.Column(name='D Total',format='F',unit='Counts',array=np.array(D)[chosen])\n",
    "    col118 = fits.Column(name='C Total',format='F',unit='Counts',array=np.array(C)[chosen])\n",
    "    col142 = fits.Column(name='Model Diff',format='F',unit='Counts',array=np.array(ModDiff)[chosen])\n",
    "    col119 = fits.Column(name='ZY Total',format='F',unit='Counts',array=np.array(ZY)[chosen])\n",
    "    col120 = fits.Column(name='ZJ Total',format='F',unit='Counts',array=np.array(ZJ)[chosen])\n",
    "    col121 = fits.Column(name='ZH Total',format='F',unit='Counts',array=np.array(ZH)[chosen])\n",
    "    col122 = fits.Column(name='ZK Total',format='F',unit='Counts',array=np.array(ZK)[chosen])\n",
    "    col123 = fits.Column(name='YJ Total',format='F',unit='Counts',array=np.array(YJ)[chosen])\n",
    "    col124 = fits.Column(name='YH Total',format='F',unit='Counts',array=np.array(YH)[chosen])\n",
    "    col125 = fits.Column(name='YK Total',format='F',unit='Counts',array=np.array(YK)[chosen])\n",
    "    col126 = fits.Column(name='JH Total',format='F',unit='Counts',array=np.array(JH)[chosen])\n",
    "    col127 = fits.Column(name='JK Total',format='F',unit='Counts',array=np.array(JK)[chosen])\n",
    "    col128 = fits.Column(name='HK Total',format='F',unit='Counts',array=np.array(HK)[chosen])\n",
    "    col129 = fits.Column(name='Filters',format='F',unit='Number',array=np.array(Filters)[chosen])\n",
    "    col130 = fits.Column(name='Total Counts',format='F',unit='Number',array=np.array(Total)[chosen])\n",
    "    col131 = fits.Column(name='Percentage Count',format='F',unit='Percentage',array=np.array(Percentage)[chosen])\n",
    "    col143 = fits.Column(name='Max_Percentage',format='F',unit='Percentage',array=np.array(Max_Perc)[chosen])\n",
    "\n",
    "    cols = fits.ColDefs([col1,col2,col3,col4,col3a,col4a,col5,col6,col7,col8,col9,col10,\n",
    "                         col11,col12,col13,col14,col15,col16,col17,col18,col19,col20,\n",
    "                         col21,col22,col23,col24,col25,col26,col27,col28,col29,col30,\n",
    "                         col31,col32,col33,col34,col35,col36,col37,col38,col39,col40,\n",
    "                         col41,col42,col43,col44,col45,col46,col47,col48,col49,col50,\n",
    "                         col51,col52,col53,col54,col55,col56,col57,col58,col59,col60,\n",
    "                         col61,col62,col63,col64,col65,col66,col67,col68,col69,col70,\n",
    "                         col71,col72,col73,col74,col75,col76,col77,col78,col79,col80,\n",
    "                         col81,col82,col83,col84,col85,col86,col87,col88,col89,col90,\n",
    "                         col91,col92,col93,col94,col95,col96,col97,col98,col99,col100,\n",
    "                         col101,col102,col103,col104,col105,col132,col106,col133,col107,\n",
    "                         col134,col108,col135,col109,col136,col110,col137,col111,col138,\n",
    "                         col112,col139,col113,col114,col140,col115,col116,col141,col117,\n",
    "                         col118,col142,col119,col120,col121,col122,col123,col124,col125,\n",
    "                         col126,col127,col128,col129,col130,col131,col143])\n",
    "    \n",
    "    '''Here we make the fits file'''\n",
    "\n",
    "    data = fits.BinTableHDU.from_columns(cols)\n",
    "    data.writeto(output_fits)\n",
    "\n",
    "    '''Now we make the reg file'''\n",
    "\n",
    "    pos_data = []\n",
    "    RA_dms = np.array(RA_dms)\n",
    "    Dec_dms = np.array(Dec_dms)\n",
    "\n",
    "    for v in range(len(chosen)):\n",
    "\n",
    "        pos_data.append([RA_dms[chosen][v],Dec_dms[chosen][v]])\n",
    "\n",
    "    with open(output_reg, 'w', newline='') as csvfile:\n",
    "        reg_file = csv.writer(csvfile, delimiter=',', quotechar=' ', quoting=csv.QUOTE_ALL)\n",
    "        for w in range(len(pos_data)):\n",
    "            reg_file.writerow([pos_data[w][0] , pos_data[w][1]])\n",
    "            \n",
    "def Distance(d=10):\n",
    "    mu = 5*math.log(d,10)-5\n",
    "    return mu"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "vlm_object_probability('0606_final.fits','model_iso.fits','0607_final')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "vlm_object_added('0606_final.fits','model_iso.fits','0607_final_added')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "mu_far = Distance(420)\n",
    "mu_near = Distance(350)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "bd_range_1myr = range(1300,2931)\n",
    "full_range = range(1300,4000)\n",
    "full_rangey = range(1300,4000)\n",
    "full_rangeo = range(1600,4000)\n",
    "full_rangec = range(1300,4000)\n",
    "full_ranged = range(1650,4000)\n",
    "\n",
    "Av = 1.1\n",
    "\n",
    "Az = 0.5006359319 * Av\n",
    "Ay = 0.3913227425 * Av\n",
    "Aj = 0.281344478 * Av\n",
    "Ah = 0.1812818241 * Av\n",
    "Ak = 0.1180678823 * Av\n",
    "\n",
    "Azy = Az - Ay\n",
    "Azj = Az - Aj\n",
    "Azh = Az - Ah\n",
    "Azk = Az - Ak\n",
    "Ayj = Ay - Aj\n",
    "Ayh = Ay - Ah\n",
    "Ayk = Ay - Ak\n",
    "Ajh = Aj - Ah\n",
    "Ajk = Aj - Ak\n",
    "Ahk = Ah - Ak"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "print(Azj)\n",
    "print(Az)\n",
    "print(mu_far)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "info = fits.open('final_3filt_wmissing_cut.fits')\n",
    "\n",
    "ra = info[1].data['RA']\n",
    "dec = info[1].data['Dec']\n",
    "ra2000 = info[1].data['RA2000']\n",
    "dec2000 = info[1].data['Dec2000']\n",
    "mag_z = info[1].data['Mag_Z']\n",
    "obs_mag_z = info[1].data['Obs_Mag_Z']\n",
    "aper_z = info[1].data['Aper_Z']\n",
    "aper_z_err = info[1].data['Aper_Z_err']\n",
    "mag_y = info[1].data['Mag_Y']\n",
    "obs_mag_y = info[1].data['Obs_Mag_Y']\n",
    "aper_y = info[1].data['Aper_Y']\n",
    "aper_y_err = info[1].data['Aper_Y_err']\n",
    "mag_j = info[1].data['Mag_J']\n",
    "obs_mag_j = info[1].data['Obs_Mag_J']\n",
    "aper_j = info[1].data['Aper_J']\n",
    "aper_j_err = info[1].data['Aper_J_err']\n",
    "mag_h = info[1].data['Mag_H']\n",
    "obs_mag_h = info[1].data['Obs_Mag_H']\n",
    "aper_h = info[1].data['Aper_H']\n",
    "aper_h_err = info[1].data['Aper_H_err']\n",
    "mag_k = info[1].data['Mag_K']\n",
    "obs_mag_k = info[1].data['Obs_Mag_K']\n",
    "aper_k = info[1].data['Aper_K']\n",
    "aper_k_err = info[1].data['Aper_K_err']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "model = fits.open('model_iso.fits')\n",
    "\n",
    "mass1 = model[1].data['Mass'][0:28]\n",
    "teff1 = model[1].data['Teff'][0:28]\n",
    "zmd1 = model[1].data['ZmagD'][0:28]\n",
    "ymd1 = model[1].data['YmagD'][0:28]\n",
    "jmd1 = model[1].data['JmagD'][0:28]\n",
    "hmd1 = model[1].data['HmagD'][0:28]\n",
    "kmd1 = model[1].data['KsmagD'][0:28]\n",
    "zmc1 = model[1].data['ZmagC'][0:28]\n",
    "ymc1 = model[1].data['YmagC'][0:28]\n",
    "jmc1 = model[1].data['JmagC'][0:28]\n",
    "hmc1 = model[1].data['HmagC'][0:28]\n",
    "kmc1 = model[1].data['KsmagC'][0:28]\n",
    "\n",
    "mass_fit = np.polyfit(teff1, mass1, 14)\n",
    "mass_poly = np.poly1d(mass_fit)\n",
    "teff_fit = np.polyfit(mass1, teff1, 14)\n",
    "teff_poly = np.poly1d(teff_fit)\n",
    "zc_fit = np.polyfit(teff1, zmc1, 14)\n",
    "zc_poly = np.poly1d(zc_fit)\n",
    "zd_fit = np.polyfit(teff1, zmd1, 14)\n",
    "zd_poly = np.poly1d(zd_fit)\n",
    "yc_fit = np.polyfit(teff1, ymc1, 14)\n",
    "yc_poly = np.poly1d(yc_fit)\n",
    "yd_fit = np.polyfit(teff1, ymd1, 14)\n",
    "yd_poly = np.poly1d(yd_fit)\n",
    "jc_fit = np.polyfit(teff1, jmc1, 14)\n",
    "jc_poly = np.poly1d(jc_fit)\n",
    "jd_fit = np.polyfit(teff1, jmd1, 14)\n",
    "jd_poly = np.poly1d(jd_fit)\n",
    "hc_fit = np.polyfit(teff1, hmc1, 14)\n",
    "hc_poly = np.poly1d(hc_fit)\n",
    "hd_fit = np.polyfit(teff1, hmd1, 14)\n",
    "hd_poly = np.poly1d(hd_fit)\n",
    "kc_fit = np.polyfit(teff1, kmc1, 14)\n",
    "kc_poly = np.poly1d(kc_fit)\n",
    "kd_fit = np.polyfit(teff1, kmd1, 14)\n",
    "kd_poly = np.poly1d(kd_fit)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "model = fits.open('model_iso.fits')\n",
    "\n",
    "mass5 = model[1].data['Mass'][28:57]\n",
    "teff5 = model[1].data['Teff'][28:57]\n",
    "zmd5 = model[1].data['ZmagD'][28:57]\n",
    "ymd5 = model[1].data['YmagD'][28:57]\n",
    "jmd5 = model[1].data['JmagD'][28:57]\n",
    "hmd5 = model[1].data['HmagD'][28:57]\n",
    "kmd5 = model[1].data['KsmagD'][28:57]\n",
    "zmc5 = model[1].data['ZmagC'][28:57]\n",
    "ymc5 = model[1].data['YmagC'][28:57]\n",
    "jmc5 = model[1].data['JmagC'][28:57]\n",
    "hmc5 = model[1].data['HmagC'][28:57]\n",
    "kmc5 = model[1].data['KsmagC'][28:57]\n",
    "\n",
    "\n",
    "masso_fit = np.polyfit(teff5, mass5, 9)\n",
    "masso_poly = np.poly1d(masso_fit)\n",
    "zco_fit = np.polyfit(teff5, zmc5, 9)\n",
    "zco_poly = np.poly1d(zco_fit)\n",
    "zdo_fit = np.polyfit(teff5, zmd5, 9)\n",
    "zdo_poly = np.poly1d(zdo_fit)\n",
    "yco_fit = np.polyfit(teff5, ymc5, 9)\n",
    "yco_poly = np.poly1d(yco_fit)\n",
    "ydo_fit = np.polyfit(teff5, ymd5, 9)\n",
    "ydo_poly = np.poly1d(ydo_fit)\n",
    "jco_fit = np.polyfit(teff5, jmc5, 9)\n",
    "jco_poly = np.poly1d(jco_fit)\n",
    "jdo_fit = np.polyfit(teff5, jmd5, 9)\n",
    "jdo_poly = np.poly1d(jdo_fit)\n",
    "hco_fit = np.polyfit(teff5, hmc5, 9)\n",
    "hco_poly = np.poly1d(hco_fit)\n",
    "hdo_fit = np.polyfit(teff5, hmd5, 9)\n",
    "hdo_poly = np.poly1d(hdo_fit)\n",
    "kco_fit = np.polyfit(teff5, kmc5, 9)\n",
    "kco_poly = np.poly1d(kco_fit)\n",
    "kdo_fit = np.polyfit(teff5, kmd5, 9)\n",
    "kdo_poly = np.poly1d(kdo_fit)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "err_z = aper_z_err/(aper_z)\n",
    "err_y = aper_y_err/(aper_y)\n",
    "err_j = aper_j_err/(aper_j)\n",
    "err_h = aper_h_err/(aper_h)\n",
    "err_k = aper_k_err/(aper_k)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "p = 8\n",
    "\n",
    "idz = np.isfinite(mag_z) & np.isfinite(err_z)\n",
    "zerr_fit = np.polyfit(mag_z[idz],err_z[idz],p)\n",
    "zerr_poly = np.poly1d(zerr_fit)\n",
    "idy = np.isfinite(mag_y) & np.isfinite(err_y)\n",
    "yerr_fit = np.polyfit(mag_y[idy],err_y[idy],p)\n",
    "yerr_poly = np.poly1d(yerr_fit)\n",
    "idj = np.isfinite(mag_j) & np.isfinite(err_j)\n",
    "jerr_fit = np.polyfit(mag_j[idj],err_j[idj],p)\n",
    "jerr_poly = np.poly1d(jerr_fit)\n",
    "idh = np.isfinite(mag_h) & np.isfinite(err_h)\n",
    "herr_fit = np.polyfit(mag_h[idh],err_h[idh],p)\n",
    "herr_poly = np.poly1d(herr_fit)\n",
    "idk = np.isfinite(mag_k) & np.isfinite(err_k)\n",
    "kerr_fit = np.polyfit(mag_k[idk],err_k[idk],p)\n",
    "kerr_poly = np.poly1d(kerr_fit)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "'''I used this cell originally to calculate the temperatures of certain magnitude sources (y)'''\n",
    "\n",
    "y = 11\n",
    "temps = []\n",
    "\n",
    "x = 1600\n",
    "while (zd_poly(x) + mu_near + zerr_poly(zd_poly(x) + mu_near)) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('1:',x)\n",
    "\n",
    "x = 1600\n",
    "while zd_poly(x) + mu_near + Az - zerr_poly(zd_poly(x) + mu_near + Az) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('2:',x)\n",
    "\n",
    "x = 1600\n",
    "while zc_poly(x) + mu_near + zerr_poly(zc_poly(x) + mu_near) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('3:',x)\n",
    "\n",
    "x = 1600\n",
    "while zc_poly(x) + mu_near + Az - zerr_poly(zc_poly(x) + mu_near + Az) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('4:',x)\n",
    "\n",
    "x = 1600\n",
    "while zd_poly(x) + mu_far + zerr_poly(zd_poly(x) + mu_far) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('5:',x)\n",
    "\n",
    "x = 1600\n",
    "while zd_poly(x) + mu_far + Az - zerr_poly(zd_poly(x) + mu_far + Az) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('6:',x)\n",
    "\n",
    "x = 1600\n",
    "while zc_poly(x) + mu_far + zerr_poly(zc_poly(x) + mu_far) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('7:',x)\n",
    "\n",
    "x = 1600\n",
    "while zc_poly(x) + mu_far + Az - zerr_poly(zc_poly(x) + mu_far + Az) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('8:',x)\n",
    "\n",
    "x = 1600\n",
    "while zdo_poly(x) + mu_near + zerr_poly(zdo_poly(x) + mu_near) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('9:',x)\n",
    "\n",
    "x = 1600\n",
    "while zdo_poly(x) + mu_near + Az - zerr_poly(zdo_poly(x) + mu_near + Az) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('10:',x)\n",
    "\n",
    "x = 1600\n",
    "while zco_poly(x) + mu_near + zerr_poly(zco_poly(x) + mu_near) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('11:',x)\n",
    "\n",
    "x = 1600\n",
    "while zco_poly(x) + mu_near + Az - zerr_poly(zco_poly(x) + mu_near + Az) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('12:',x)\n",
    "\n",
    "x = 1600\n",
    "while zdo_poly(x) + mu_far + zerr_poly(zdo_poly(x) + mu_far) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('13:',x)\n",
    "\n",
    "x = 1600\n",
    "while zdo_poly(x) + mu_far + Az - zerr_poly(zdo_poly(x) + mu_far + Az) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('14:',x)\n",
    "\n",
    "x = 1600\n",
    "while zco_poly(x) + mu_far + zerr_poly(zco_poly(x) + mu_far) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('15:',x)\n",
    "\n",
    "x = 1600\n",
    "while zco_poly(x) + mu_far + Az - zerr_poly(zco_poly(x) + mu_far + Az) > y and x < 4500:\n",
    "    x = x+1    \n",
    "if x!=1600:\n",
    "    temps.append(x)\n",
    "    print('16:',x)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "brown_tempy = 2931.6\n",
    "brown_tempo = 2971.6\n",
    "tempy = 2306.4\n",
    "tempo = 2068.6"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "'''I used the border temperatures found just above to then determine the corresponding magnitudes in each set-up.'''\n",
    "ynd_l_x = zd_poly(tempy) - jd_poly(tempy) - np.sqrt(zerr_poly(zd_poly(tempy) + mu_near)**2 + jerr_poly(jd_poly(tempy) + mu_near)**2)\n",
    "ynd_l_y = zd_poly(tempy) + mu_near + zerr_poly(zd_poly(tempy) + mu_near)\n",
    "ynd_r_x = zd_poly(tempy) - jd_poly(tempy) + Azj + np.sqrt(zerr_poly(zd_poly(tempy) + mu_near + Az)**2 + jerr_poly(jd_poly(tempy) + mu_near + Aj)**2)\n",
    "ynd_r_y = zd_poly(tempy) + mu_near + Az - zerr_poly(zd_poly(tempy) + mu_near + Az)\n",
    "ync_l_x = zc_poly(tempy) - jc_poly(tempy) - np.sqrt(zerr_poly(zc_poly(tempy) + mu_near)**2 + jerr_poly(jc_poly(tempy) + mu_near)**2)\n",
    "ync_l_y = zc_poly(tempy) + mu_near + zerr_poly(zc_poly(tempy) + mu_near)\n",
    "ync_r_x = zc_poly(tempy) - jc_poly(tempy) + Azj + np.sqrt(zerr_poly(zc_poly(tempy) + mu_near + Az)**2 + jerr_poly(jc_poly(tempy) + mu_near + Aj)**2)\n",
    "ync_r_y = zc_poly(tempy) + mu_near + Az - zerr_poly(zc_poly(tempy) + mu_near + Az)\n",
    "yfd_l_x = zd_poly(tempy) - jd_poly(tempy) - np.sqrt(zerr_poly(zd_poly(tempy) + mu_far)**2 + jerr_poly(jd_poly(tempy) + mu_far)**2)\n",
    "yfd_l_y = zd_poly(tempy) + mu_far + zerr_poly(zd_poly(tempy) + mu_far)\n",
    "yfd_r_x = zd_poly(tempy) - jd_poly(tempy) + Azj + np.sqrt(zerr_poly(zd_poly(tempy) + mu_far + Az)**2 + jerr_poly(jd_poly(tempy) + mu_far + Aj)**2)\n",
    "yfd_r_y = zd_poly(tempy) + mu_far + Az - zerr_poly(zd_poly(tempy) + mu_far + Az)\n",
    "yfc_l_x = zc_poly(tempy) - jc_poly(tempy) - np.sqrt(zerr_poly(zc_poly(tempy) + mu_far)**2 + jerr_poly(jc_poly(tempy) + mu_far)**2)\n",
    "yfc_l_y = zc_poly(tempy) + mu_far + zerr_poly(zc_poly(tempy) + mu_far)\n",
    "yfc_r_x = zc_poly(tempy) - jc_poly(tempy) + Azj + np.sqrt(zerr_poly(zc_poly(tempy) + mu_far + Az)**2 + jerr_poly(jc_poly(tempy) + mu_far + Aj)**2)\n",
    "yfc_r_y = zc_poly(tempy) + mu_far + Az - zerr_poly(zc_poly(tempy) + mu_far + Az)\n",
    "ond_l_x = zdo_poly(tempo) - jdo_poly(tempo) - np.sqrt(zerr_poly(zdo_poly(tempo) + mu_near)**2 + jerr_poly(jdo_poly(tempo) + mu_near)**2)\n",
    "ond_l_y = zdo_poly(tempo) + mu_near + zerr_poly(zdo_poly(tempo) + mu_near)\n",
    "ond_r_x = zdo_poly(tempo) - jdo_poly(tempo) + Azj + np.sqrt(zerr_poly(zdo_poly(tempo) + mu_near + Az)**2 + jerr_poly(jdo_poly(tempo) + mu_near + Aj)**2)\n",
    "ond_r_y = zdo_poly(tempo) + mu_near + Az - zerr_poly(zdo_poly(tempo) + mu_near + Az)\n",
    "onc_l_x = zco_poly(tempo) - jco_poly(tempo) - np.sqrt(zerr_poly(zco_poly(tempo) + mu_near)**2 + jerr_poly(jco_poly(tempo) + mu_near)**2)\n",
    "onc_l_y = zco_poly(tempo) + mu_near + zerr_poly(zco_poly(tempo) + mu_near)\n",
    "onc_r_x = zco_poly(tempo) - jco_poly(tempo) + Azj + np.sqrt(zerr_poly(zco_poly(tempo) + mu_near + Az)**2 + jerr_poly(jco_poly(tempo) + mu_near + Aj)**2)\n",
    "onc_r_y = zco_poly(tempo) + mu_near + Az - zerr_poly(zco_poly(tempo) + mu_near + Az)\n",
    "ofd_l_x = zdo_poly(tempo) - jdo_poly(tempo) - np.sqrt(zerr_poly(zdo_poly(tempo) + mu_far)**2 + jerr_poly(jdo_poly(tempo) + mu_far)**2)\n",
    "ofd_l_y = zdo_poly(tempo) + mu_far + zerr_poly(zdo_poly(tempo) + mu_far)\n",
    "ofd_r_x = zdo_poly(tempo) - jdo_poly(tempo) + Azj + np.sqrt(zerr_poly(zdo_poly(tempo) + mu_far + Az)**2 + jerr_poly(jdo_poly(tempo) + mu_far + Aj)**2)\n",
    "ofd_r_y = zdo_poly(tempo) + mu_far + Az - zerr_poly(zdo_poly(tempo) + mu_far + Az)\n",
    "ofc_l_x = zco_poly(tempo) - jco_poly(tempo) - np.sqrt(zerr_poly(zco_poly(tempo) + mu_far)**2 + jerr_poly(jco_poly(tempo) + mu_far)**2)\n",
    "ofc_l_y = zco_poly(tempo) + mu_far + zerr_poly(zco_poly(tempo) + mu_far)\n",
    "ofc_r_x = zco_poly(tempo) - jco_poly(tempo) + Azj + np.sqrt(zerr_poly(zco_poly(tempo) + mu_far + Az)**2 + jerr_poly(jco_poly(tempo) + mu_far + Aj)**2)\n",
    "ofc_r_y = zco_poly(tempo) + mu_far + Az - zerr_poly(zco_poly(tempo) + mu_far + Az)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "ynd_l_x_brown = zd_poly(brown_tempy) - jd_poly(brown_tempy) - np.sqrt(zerr_poly(zd_poly(brown_tempy) + mu_near)**2 + jerr_poly(jd_poly(brown_tempy) + mu_near)**2)\n",
    "ynd_l_y_brown = zd_poly(brown_tempy) + mu_near + zerr_poly(zd_poly(brown_tempy) + mu_near)\n",
    "ynd_r_x_brown = zd_poly(brown_tempy) - jd_poly(brown_tempy) + Azj + np.sqrt(zerr_poly(zd_poly(brown_tempy) + mu_near + Az)**2 + jerr_poly(jd_poly(brown_tempy) + mu_near + Aj)**2)\n",
    "ynd_r_y_brown = zd_poly(brown_tempy) + mu_near + Az - zerr_poly(zd_poly(brown_tempy) + mu_near + Az)\n",
    "ync_l_x_brown = zc_poly(brown_tempy) - jc_poly(brown_tempy) - np.sqrt(zerr_poly(zc_poly(brown_tempy) + mu_near)**2 + jerr_poly(jc_poly(brown_tempy) + mu_near)**2)\n",
    "ync_l_y_brown = zc_poly(brown_tempy) + mu_near + zerr_poly(zc_poly(brown_tempy) + mu_near)\n",
    "ync_r_x_brown = zc_poly(brown_tempy) - jc_poly(brown_tempy) + Azj + np.sqrt(zerr_poly(zc_poly(brown_tempy) + mu_near + Az)**2 + jerr_poly(jc_poly(brown_tempy) + mu_near + Aj)**2)\n",
    "ync_r_y_brown = zc_poly(brown_tempy) + mu_near + Az - zerr_poly(zc_poly(brown_tempy) + mu_near + Az)\n",
    "yfd_l_x_brown = zd_poly(brown_tempy) - jd_poly(brown_tempy) - np.sqrt(zerr_poly(zd_poly(brown_tempy) + mu_far)**2 + jerr_poly(jd_poly(brown_tempy) + mu_far)**2)\n",
    "yfd_l_y_brown = zd_poly(brown_tempy) + mu_far + zerr_poly(zd_poly(brown_tempy) + mu_far)\n",
    "yfd_r_x_brown = zd_poly(brown_tempy) - jd_poly(brown_tempy) + Azj + np.sqrt(zerr_poly(zd_poly(brown_tempy) + mu_far + Az)**2 + jerr_poly(jd_poly(brown_tempy) + mu_far + Aj)**2)\n",
    "yfd_r_y_brown = zd_poly(brown_tempy) + mu_far + Az - zerr_poly(zd_poly(brown_tempy) + mu_far + Az)\n",
    "yfc_l_x_brown = zc_poly(brown_tempy) - jc_poly(brown_tempy) - np.sqrt(zerr_poly(zc_poly(brown_tempy) + mu_far)**2 + jerr_poly(jc_poly(brown_tempy) + mu_far)**2)\n",
    "yfc_l_y_brown = zc_poly(brown_tempy) + mu_far + zerr_poly(zc_poly(brown_tempy) + mu_far)\n",
    "yfc_r_x_brown = zc_poly(brown_tempy) - jc_poly(brown_tempy) + Azj + np.sqrt(zerr_poly(zc_poly(brown_tempy) + mu_far + Az)**2 + jerr_poly(jc_poly(brown_tempy) + mu_far + Aj)**2)\n",
    "yfc_r_y_brown = zc_poly(brown_tempy) + mu_far + Az - zerr_poly(zc_poly(brown_tempy) + mu_far + Az)\n",
    "ond_l_x_brown = zdo_poly(brown_tempo) - jdo_poly(brown_tempo) - np.sqrt(zerr_poly(zdo_poly(brown_tempo) + mu_near)**2 + jerr_poly(jdo_poly(brown_tempo) + mu_near)**2)\n",
    "ond_l_y_brown = zdo_poly(brown_tempo) + mu_near + zerr_poly(zdo_poly(brown_tempo) + mu_near)\n",
    "ond_r_x_brown = zdo_poly(brown_tempo) - jdo_poly(brown_tempo) + Azj + np.sqrt(zerr_poly(zdo_poly(brown_tempo) + mu_near + Az)**2 + jerr_poly(jdo_poly(brown_tempo) + mu_near + Aj)**2)\n",
    "ond_r_y_brown = zdo_poly(brown_tempo) + mu_near + Az - zerr_poly(zdo_poly(brown_tempo) + mu_near + Az)\n",
    "onc_l_x_brown = zco_poly(brown_tempo) - jco_poly(brown_tempo) - np.sqrt(zerr_poly(zco_poly(brown_tempo) + mu_near)**2 + jerr_poly(jco_poly(brown_tempo) + mu_near)**2)\n",
    "onc_l_y_brown = zco_poly(brown_tempo) + mu_near + zerr_poly(zco_poly(brown_tempo) + mu_near)\n",
    "onc_r_x_brown = zco_poly(brown_tempo) - jco_poly(brown_tempo) + Azj + np.sqrt(zerr_poly(zco_poly(brown_tempo) + mu_near + Az)**2 + jerr_poly(jco_poly(brown_tempo) + mu_near + Aj)**2)\n",
    "onc_r_y_brown = zco_poly(brown_tempo) + mu_near + Az - zerr_poly(zco_poly(brown_tempo) + mu_near + Az)\n",
    "ofd_l_x_brown = zdo_poly(brown_tempo) - jdo_poly(brown_tempo) - np.sqrt(zerr_poly(zdo_poly(brown_tempo) + mu_far)**2 + jerr_poly(jdo_poly(brown_tempo) + mu_far)**2)\n",
    "ofd_l_y_brown = zdo_poly(brown_tempo) + mu_far + zerr_poly(zdo_poly(brown_tempo) + mu_far)\n",
    "ofd_r_x_brown = zdo_poly(brown_tempo) - jdo_poly(brown_tempo) + Azj + np.sqrt(zerr_poly(zdo_poly(brown_tempo) + mu_far + Az)**2 + jerr_poly(jdo_poly(brown_tempo) + mu_far + Aj)**2)\n",
    "ofd_r_y_brown = zdo_poly(brown_tempo) + mu_far + Az - zerr_poly(zdo_poly(brown_tempo) + mu_far + Az)\n",
    "ofc_l_x_brown = zco_poly(brown_tempo) - jco_poly(brown_tempo) - np.sqrt(zerr_poly(zco_poly(brown_tempo) + mu_far)**2 + jerr_poly(jco_poly(brown_tempo) + mu_far)**2)\n",
    "ofc_l_y_brown = zco_poly(brown_tempo) + mu_far + zerr_poly(zco_poly(brown_tempo) + mu_far)\n",
    "ofc_r_x_brown = zco_poly(brown_tempo) - jco_poly(brown_tempo) + Azj + np.sqrt(zerr_poly(zco_poly(brown_tempo) + mu_far + Az)**2 + jerr_poly(jco_poly(brown_tempo) + mu_far + Aj)**2)\n",
    "ofc_r_y_brown = zco_poly(brown_tempo) + mu_far + Az - zerr_poly(zco_poly(brown_tempo) + mu_far + Az)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "ipmo_mag = (ynd_l_y,ynd_r_y,ync_l_y,ync_r_y,yfd_l_y,yfd_r_y,yfc_l_y,yfc_r_y,ond_l_y,ond_r_y,onc_l_y,onc_r_y,ofd_l_y,ofd_r_y,ofc_l_y,ofc_r_y)\n",
    "print(max(ipmo_mag))\n",
    "print(min(ipmo_mag))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "bd_mag = (ynd_l_y_brown,ynd_r_y_brown,ync_l_y_brown,ync_r_y_brown,yfd_l_y_brown,yfd_r_y_brown,yfc_l_y_brown,yfc_r_y_brown,ond_l_y_brown,ond_r_y_brown,onc_l_y_brown,onc_r_y_brown,ofd_l_y_brown,ofd_r_y_brown,ofc_l_y_brown,ofc_r_y_brown)\n",
    "print(max(bd_mag))\n",
    "print(min(bd_mag))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "'An example of how the error affects the main model line'''\n",
    "plt.figure(figsize=(40,20))\n",
    "plt.plot(mag_z-mag_j,mag_z,'.')\n",
    "plt.plot(np.sqrt(zerr_poly(zc_poly(full_range)+8.074486080165673+Az)**2+jerr_poly(jc_poly(full_range)+8.074486080165673+Aj)**2),zc_poly(full_range)+8.074486080165673+0.5006359319)\n",
    "plt.plot((zc_poly(full_range)-jc_poly(full_range)+(1*0.2192914539)),zc_poly(full_range)+8.074486080165673+0.5006359319)\n",
    "#plt.plot((zc_poly(full_range)-jc_poly(full_range)+(0*0.2192914539))-np.sqrt(zerr_poly(zc_poly(full_range)+8.074486080165673+0.5006359319)),zc_poly(full_range)+8.074486080165673+0.5006359319)\n",
    "plt.plot((zc_poly(full_range)-jc_poly(full_range)+(1*0.2192914539))-np.sqrt(zerr_poly(zc_poly(full_range)+8.074486080165673+Az)**2+jerr_poly(jc_poly(full_range)+8.074486080165673+Aj)**2),zc_poly(full_range)+8.074486080165673+0.5006359319)\n",
    "plt.plot((zc_poly(full_range)-jc_poly(full_range)+(1*0.2192914539))+np.sqrt(zerr_poly(zc_poly(full_range)+8.074486080165673+Az)**2+jerr_poly(jc_poly(full_range)+8.074486080165673+Aj)**2),zc_poly(full_range)+8.074486080165673+0.5006359319)\n",
    "plt.gca().invert_yaxis()\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "'''This one shows extinction effects'''\n",
    "plt.figure(figsize=(40,20))\n",
    "plt.plot(mag_z-mag_j,mag_z,'.')\n",
    "plt.plot((zc_poly(full_range)-jc_poly(full_range)+(0*0.2192914539)),zc_poly(full_range)+8.074486080165673+0*0.5006359319)\n",
    "plt.plot((zc_poly(full_range)-jc_poly(full_range)+(1*0.2192914539)),zc_poly(full_range)+8.074486080165673+1*0.5006359319)\n",
    "plt.plot((zc_poly(full_range)-jc_poly(full_range)+(2*0.2192914539)),zc_poly(full_range)+8.074486080165673+2*0.5006359319)\n",
    "#plt.ylim(21,24)\n",
    "plt.gca().invert_yaxis()\n",
    "#plt.xlim(1.5,3.5)\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "bd_range_1myr = range(1300,2931)\n",
    "full_range = range(1300,4000)\n",
    "full_rangey = range(1300,4000)\n",
    "full_rangeo = range(1600,4000)\n",
    "full_rangec = range(1300,4000)\n",
    "full_ranged = range(1650,4000)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "'''This and the next 9 figures are for the various CMD combinations, with model lines'''\n",
    "plt.figure(figsize=(40,20))\n",
    "plt.rc('xtick',labelsize=28)\n",
    "plt.rc('ytick',labelsize=28)\n",
    "plt.plot(mag_z-mag_y,mag_z,'.',color='grey')\n",
    "plt.plot(zd_poly(full_ranged)-yd_poly(full_ranged) - np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_near)**2 + yerr_poly(yd_poly(full_ranged) + mu_near)**2),zd_poly(full_ranged) + mu_near + zerr_poly(zd_poly(full_ranged) + mu_near),color = 'C0',label='1 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zd_poly(full_ranged)-yd_poly(full_ranged) + Azy + np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_near + Az)**2 + yerr_poly(yd_poly(full_ranged) + mu_near + Ay)**2), zd_poly(full_ranged) + mu_near + Az - zerr_poly(zd_poly(full_ranged) + mu_near + Az),color = 'C0',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-yc_poly(full_rangec) - np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_near)**2 + yerr_poly(yc_poly(full_rangec) + mu_near)**2),zc_poly(full_rangec) + mu_near + zerr_poly(zc_poly(full_rangec) + mu_near),color = 'C1',label='1 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-yc_poly(full_rangec) + Azy + np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_near + Az)**2 + yerr_poly(yc_poly(full_rangec) + mu_near + Ay)**2) , zc_poly(full_rangec) + mu_near + Az - zerr_poly(zc_poly(full_rangec) + mu_near + Az),color = 'C1',linewidth=4.0)\n",
    "plt.plot(zd_poly(full_ranged)-yd_poly(full_ranged) - np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_far)**2 + yerr_poly(yd_poly(full_ranged) + mu_far)**2),zd_poly(full_ranged) + mu_far + zerr_poly(zd_poly(full_ranged) + mu_far),color = 'C2',label='1 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zd_poly(full_ranged)-yd_poly(full_ranged) + Azy + np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_far + Az)**2 + yerr_poly(yd_poly(full_ranged) + mu_far + Ay)**2), zd_poly(full_ranged) + mu_far + Az - zerr_poly(zd_poly(full_ranged) + mu_far + Az),color = 'C2',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-yc_poly(full_rangec) - np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_far)**2 + yerr_poly(yc_poly(full_rangec) + mu_far)**2),zc_poly(full_rangec) + mu_far + zerr_poly(zc_poly(full_rangec) + mu_far),color = 'C3',label='1 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-yc_poly(full_rangec) + Azy + np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_far + Az)**2 + yerr_poly(yc_poly(full_rangec) + mu_far + Ay)**2), zc_poly(full_rangec) + mu_far + Az - zerr_poly(zc_poly(full_rangec) + mu_far + Az),color = 'C3',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-ydo_poly(full_ranged) - np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_near)**2 + yerr_poly(ydo_poly(full_ranged) + mu_near)**2),zdo_poly(full_ranged) + mu_near + zerr_poly(zdo_poly(full_ranged) + mu_near),color = 'C4',label='5 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-ydo_poly(full_ranged) + Azy + np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_near + Az)**2 + yerr_poly(ydo_poly(full_ranged) + mu_near + Ay)**2), zdo_poly(full_ranged) + mu_near + Az - zerr_poly(zdo_poly(full_ranged) + mu_near + Az),color = 'C4',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-yco_poly(full_rangec) - np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_near)**2 + yerr_poly(yco_poly(full_rangec) + mu_near)**2),zco_poly(full_rangec) + mu_near + zerr_poly(zco_poly(full_rangec) + mu_near),color = 'C5',label='5 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-yco_poly(full_rangec) + Azy + np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_near + Az)**2 + yerr_poly(yco_poly(full_rangec) + mu_near + Ay)**2), zco_poly(full_rangec) + mu_near + Az - zerr_poly(zco_poly(full_rangec) + mu_near + Az),color = 'C5',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-ydo_poly(full_ranged) - np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_far)**2 + yerr_poly(ydo_poly(full_ranged) + mu_far)**2),zdo_poly(full_ranged) + mu_far + zerr_poly(zdo_poly(full_ranged) + mu_far),color = 'C6',label='5 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-ydo_poly(full_ranged) + Azy + np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_far + Az)**2 + yerr_poly(ydo_poly(full_ranged) + mu_far + Ay)**2), zdo_poly(full_ranged) + mu_far + Az - zerr_poly(zdo_poly(full_ranged) + mu_far + Az),color = 'C6',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-yco_poly(full_rangec) - np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_far)**2 + yerr_poly(yco_poly(full_rangec) + mu_far)**2),zco_poly(full_rangec) + mu_far + zerr_poly(zco_poly(full_rangec) + mu_far),color = 'C8',label='5 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-yco_poly(full_rangec) + Azy + np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_far + Az)**2 + yerr_poly(yco_poly(full_rangec) + mu_far + Ay)**2), zco_poly(full_rangec) + mu_far + Az - zerr_poly(zco_poly(full_rangec) + mu_far + Az),color = 'C8',linewidth=4.0)\n",
    "plt.arrow(1.5,18,Azy,Az,width=0.01,length_includes_head=True,color='r')\n",
    "plt.text(1.5,17.9,'$A_v$ = 1.1',fontsize=30)\n",
    "plt.ylim(13,24)\n",
    "plt.xlim(-0.5,2.5)\n",
    "#plt.ylim(18.75,21)\n",
    "#plt.xlim(2,2.5)\n",
    "plt.legend(fontsize=32)\n",
    "plt.gca().invert_yaxis()\n",
    "plt.xlabel('Z-Y Mag', fontsize=40)\n",
    "plt.ylabel('Z Mag', fontsize=40)\n",
    "plt.savefig('cmd_zy.png')\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "plt.figure(figsize=(40,20))\n",
    "plt.rc('xtick',labelsize=28)\n",
    "plt.rc('ytick',labelsize=28)\n",
    "plt.plot(mag_z-mag_j,mag_z,'.',color='grey')\n",
    "plt.plot(zd_poly(full_ranged)-jd_poly(full_ranged) - np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_near)**2 + jerr_poly(jd_poly(full_ranged) + mu_near)**2),zd_poly(full_ranged) + mu_near + zerr_poly(zd_poly(full_ranged) + mu_near),color = 'C0',label='1 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zd_poly(full_ranged)-jd_poly(full_ranged) + Azj + np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_near + Az)**2 + jerr_poly(jd_poly(full_ranged) + mu_near + Aj)**2), zd_poly(full_ranged) + mu_near + Az - zerr_poly(zd_poly(full_ranged) + mu_near + Az),color = 'C0',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-jc_poly(full_rangec) - np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_near)**2 + jerr_poly(jc_poly(full_rangec) + mu_near)**2),zc_poly(full_rangec) + mu_near + zerr_poly(zc_poly(full_rangec) + mu_near),color = 'C1',label='1 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-jc_poly(full_rangec) + Azj + np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_near + Az)**2 + jerr_poly(jc_poly(full_rangec) + mu_near + Aj)**2) , zc_poly(full_rangec) + mu_near + Az - zerr_poly(zc_poly(full_rangec) + mu_near + Az),color = 'C1',linewidth=4.0)\n",
    "plt.plot(zd_poly(full_ranged)-jd_poly(full_ranged) - np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_far)**2 + jerr_poly(jd_poly(full_ranged) + mu_far)**2),zd_poly(full_ranged) + mu_far + zerr_poly(zd_poly(full_ranged) + mu_far),color = 'C2',label='1 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zd_poly(full_ranged)-jd_poly(full_ranged) + Azj + np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_far + Az)**2 + jerr_poly(jd_poly(full_ranged) + mu_far + Aj)**2), zd_poly(full_ranged) + mu_far + Az - zerr_poly(zd_poly(full_ranged) + mu_far + Az),color = 'C2',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-jc_poly(full_rangec) - np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_far)**2 + jerr_poly(jc_poly(full_rangec) + mu_far)**2),zc_poly(full_rangec) + mu_far + zerr_poly(zc_poly(full_rangec) + mu_far),color = 'C3',label='1 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-jc_poly(full_rangec) + Azj + np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_far + Az)**2 + jerr_poly(jc_poly(full_rangec) + mu_far + Aj)**2), zc_poly(full_rangec) + mu_far + Az - zerr_poly(zc_poly(full_rangec) + mu_far + Az),color = 'C3',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-jdo_poly(full_ranged) - np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_near)**2 + jerr_poly(jdo_poly(full_ranged) + mu_near)**2),zdo_poly(full_ranged) + mu_near + zerr_poly(zdo_poly(full_ranged) + mu_near),color = 'C4',label='5 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-jdo_poly(full_ranged) + Azj + np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_near + Az)**2 + jerr_poly(jdo_poly(full_ranged) + mu_near + Aj)**2), zdo_poly(full_ranged) + mu_near + Az - zerr_poly(zdo_poly(full_ranged) + mu_near + Az),color = 'C4',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-jco_poly(full_rangec) - np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_near)**2 + jerr_poly(jco_poly(full_rangec) + mu_near)**2),zco_poly(full_rangec) + mu_near + zerr_poly(zco_poly(full_rangec) + mu_near),color = 'C5',label='5 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-jco_poly(full_rangec) + Azj + np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_near + Az)**2 + jerr_poly(jco_poly(full_rangec) + mu_near + Aj)**2), zco_poly(full_rangec) + mu_near + Az - zerr_poly(zco_poly(full_rangec) + mu_near + Az),color = 'C5',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-jdo_poly(full_ranged) - np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_far)**2 + jerr_poly(jdo_poly(full_ranged) + mu_far)**2),zdo_poly(full_ranged) + mu_far + zerr_poly(zdo_poly(full_ranged) + mu_far),color = 'C6',label='5 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-jdo_poly(full_ranged) + Azj + np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_far + Az)**2 + jerr_poly(jdo_poly(full_ranged) + mu_far + Aj)**2), zdo_poly(full_ranged) + mu_far + Az - zerr_poly(zdo_poly(full_ranged) + mu_far + Az),color = 'C6',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-jco_poly(full_rangec) - np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_far)**2 + jerr_poly(jco_poly(full_rangec) + mu_far)**2),zco_poly(full_rangec) + mu_far + zerr_poly(zco_poly(full_rangec) + mu_far),color = 'C8',label='5 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-jco_poly(full_rangec) + Azj + np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_far + Az)**2 + jerr_poly(jco_poly(full_rangec) + mu_far + Aj)**2), zco_poly(full_rangec) + mu_far + Az - zerr_poly(zco_poly(full_rangec) + mu_far + Az),color = 'C8',linewidth=4.0)\n",
    "plt.arrow(2.5,18,Azj,Az,width=0.01,length_includes_head=True,color='r')\n",
    "plt.text(2.5,17.9,'$A_v$ = 1.1',fontsize=30)\n",
    "plt.ylim(13,24)\n",
    "plt.xlim(0,4)\n",
    "#plt.ylim(18.75,21)\n",
    "#plt.xlim(2,2.5)\n",
    "plt.legend(fontsize=32)\n",
    "plt.gca().invert_yaxis()\n",
    "plt.xlabel('Z-J Mag', fontsize=40)\n",
    "plt.ylabel('Z Mag', fontsize=40)\n",
    "plt.savefig('cmd_zj.png')\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "plt.figure(figsize=(40,20))\n",
    "plt.rc('xtick',labelsize=28)\n",
    "plt.rc('ytick',labelsize=28)\n",
    "plt.plot(mag_z-mag_h,mag_z,'.',color='grey')\n",
    "plt.plot(zd_poly(full_ranged)-hd_poly(full_ranged) - np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_near)**2 + herr_poly(hd_poly(full_ranged) + mu_near)**2),zd_poly(full_ranged) + mu_near + zerr_poly(zd_poly(full_ranged) + mu_near),color = 'C0',label='1 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zd_poly(full_ranged)-hd_poly(full_ranged) + Azh + np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_near + Az)**2 + herr_poly(hd_poly(full_ranged) + mu_near + Ah)**2), zd_poly(full_ranged) + mu_near + Az - zerr_poly(zd_poly(full_ranged) + mu_near + Az),color = 'C0',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-hc_poly(full_rangec) - np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_near)**2 + herr_poly(hc_poly(full_rangec) + mu_near)**2),zc_poly(full_rangec) + mu_near + zerr_poly(zc_poly(full_rangec) + mu_near),color = 'C1',label='1 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-hc_poly(full_rangec) + Azh + np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_near + Az)**2 + herr_poly(hc_poly(full_rangec) + mu_near + Ah)**2) , zc_poly(full_rangec) + mu_near + Az - zerr_poly(zc_poly(full_rangec) + mu_near + Az),color = 'C1',linewidth=4.0)\n",
    "plt.plot(zd_poly(full_ranged)-hd_poly(full_ranged) - np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_far)**2 + herr_poly(hd_poly(full_ranged) + mu_far)**2),zd_poly(full_ranged) + mu_far + zerr_poly(zd_poly(full_ranged) + mu_far),color = 'C2',label='1 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zd_poly(full_ranged)-hd_poly(full_ranged) + Azh + np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_far + Az)**2 + herr_poly(hd_poly(full_ranged) + mu_far + Ah)**2), zd_poly(full_ranged) + mu_far + Az - zerr_poly(zd_poly(full_ranged) + mu_far + Az),color = 'C2',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-hc_poly(full_rangec) - np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_far)**2 + herr_poly(hc_poly(full_rangec) + mu_far)**2),zc_poly(full_rangec) + mu_far + zerr_poly(zc_poly(full_rangec) + mu_far),color = 'C3',label='1 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-hc_poly(full_rangec) + Azh + np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_far + Az)**2 + herr_poly(hc_poly(full_rangec) + mu_far + Ah)**2), zc_poly(full_rangec) + mu_far + Az - zerr_poly(zc_poly(full_rangec) + mu_far + Az),color = 'C3',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-hdo_poly(full_ranged) - np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_near)**2 + herr_poly(hdo_poly(full_ranged) + mu_near)**2),zdo_poly(full_ranged) + mu_near + zerr_poly(zdo_poly(full_ranged) + mu_near),color = 'C4',label='5 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-hdo_poly(full_ranged) + Azh + np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_near + Az)**2 + herr_poly(hdo_poly(full_ranged) + mu_near + Ah)**2), zdo_poly(full_ranged) + mu_near + Az - zerr_poly(zdo_poly(full_ranged) + mu_near + Az),color = 'C4',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-hco_poly(full_rangec) - np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_near)**2 + herr_poly(hco_poly(full_rangec) + mu_near)**2),zco_poly(full_rangec) + mu_near + zerr_poly(zco_poly(full_rangec) + mu_near),color = 'C5',label='5 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-hco_poly(full_rangec) + Azh + np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_near + Az)**2 + herr_poly(hco_poly(full_rangec) + mu_near + Ah)**2), zco_poly(full_rangec) + mu_near + Az - zerr_poly(zco_poly(full_rangec) + mu_near + Az),color = 'C5',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-hdo_poly(full_ranged) - np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_far)**2 + herr_poly(hdo_poly(full_ranged) + mu_far)**2),zdo_poly(full_ranged) + mu_far + zerr_poly(zdo_poly(full_ranged) + mu_far),color = 'C6',label='5 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-hdo_poly(full_ranged) + Azh + np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_far + Az)**2 + herr_poly(hdo_poly(full_ranged) + mu_far + Ah)**2), zdo_poly(full_ranged) + mu_far + Az - zerr_poly(zdo_poly(full_ranged) + mu_far + Az),color = 'C6',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-hco_poly(full_rangec) - np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_far)**2 + herr_poly(hco_poly(full_rangec) + mu_far)**2),zco_poly(full_rangec) + mu_far + zerr_poly(zco_poly(full_rangec) + mu_far),color = 'C8',label='5 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-hco_poly(full_rangec) + Azh + np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_far + Az)**2 + herr_poly(hco_poly(full_rangec) + mu_far + Ah)**2), zco_poly(full_rangec) + mu_far + Az - zerr_poly(zco_poly(full_rangec) + mu_far + Az),color = 'C8',linewidth=4.0)\n",
    "plt.arrow(3,18,Azh,Az,width=0.01,length_includes_head=True,color='r')\n",
    "plt.text(3,17.9,'$A_v$ = 1.1',fontsize=30)\n",
    "plt.ylim(13,24)\n",
    "plt.xlim(0.5,4.25)\n",
    "#plt.ylim(18.75,21)\n",
    "#plt.xlim(2,2.5)\n",
    "plt.legend(fontsize=32)\n",
    "plt.gca().invert_yaxis()\n",
    "plt.xlabel('Z-H Mag', fontsize=40)\n",
    "plt.ylabel('Z Mag', fontsize=40)\n",
    "plt.savefig('cmd_zh.png')\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "plt.figure(figsize=(40,20))\n",
    "plt.rc('xtick',labelsize=28)\n",
    "plt.rc('ytick',labelsize=28)\n",
    "plt.plot(mag_z-mag_k,mag_z,'.',color='grey')\n",
    "plt.plot(zd_poly(full_ranged)-kd_poly(full_ranged) - np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_near)**2 + kerr_poly(kd_poly(full_ranged) + mu_near)**2),zd_poly(full_ranged) + mu_near + zerr_poly(zd_poly(full_ranged) + mu_near),color = 'C0',label='1 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zd_poly(full_ranged)-kd_poly(full_ranged) + Azk + np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_near + Az)**2 + kerr_poly(kd_poly(full_ranged) + mu_near + Ak)**2), zd_poly(full_ranged) + mu_near + Az - zerr_poly(zd_poly(full_ranged) + mu_near + Az),color = 'C0',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-kc_poly(full_rangec) - np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_near)**2 + kerr_poly(kc_poly(full_rangec) + mu_near)**2),zc_poly(full_rangec) + mu_near + zerr_poly(zc_poly(full_rangec) + mu_near),color = 'C1',label='1 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-kc_poly(full_rangec) + Azk + np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_near + Az)**2 + kerr_poly(kc_poly(full_rangec) + mu_near + Ak)**2) , zc_poly(full_rangec) + mu_near + Az - zerr_poly(zc_poly(full_rangec) + mu_near + Az),color = 'C1',linewidth=4.0)\n",
    "plt.plot(zd_poly(full_ranged)-kd_poly(full_ranged) - np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_far)**2 + kerr_poly(kd_poly(full_ranged) + mu_far)**2),zd_poly(full_ranged) + mu_far + zerr_poly(zd_poly(full_ranged) + mu_far),color = 'C2',label='1 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zd_poly(full_ranged)-kd_poly(full_ranged) + Azk + np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_far + Az)**2 + kerr_poly(kd_poly(full_ranged) + mu_far + Ak)**2), zd_poly(full_ranged) + mu_far + Az - zerr_poly(zd_poly(full_ranged) + mu_far + Az),color = 'C2',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-kc_poly(full_rangec) - np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_far)**2 + kerr_poly(kc_poly(full_rangec) + mu_far)**2),zc_poly(full_rangec) + mu_far + zerr_poly(zc_poly(full_rangec) + mu_far),color = 'C3',label='1 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zc_poly(full_rangec)-kc_poly(full_rangec) + Azk + np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_far + Az)**2 + kerr_poly(kc_poly(full_rangec) + mu_far + Ak)**2), zc_poly(full_rangec) + mu_far + Az - zerr_poly(zc_poly(full_rangec) + mu_far + Az),color = 'C3',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-kdo_poly(full_ranged) - np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_near)**2 + kerr_poly(kdo_poly(full_ranged) + mu_near)**2),zdo_poly(full_ranged) + mu_near + zerr_poly(zdo_poly(full_ranged) + mu_near),color = 'C4',label='5 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-kdo_poly(full_ranged) + Azk + np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_near + Az)**2 + kerr_poly(kdo_poly(full_ranged) + mu_near + Ak)**2), zdo_poly(full_ranged) + mu_near + Az - zerr_poly(zdo_poly(full_ranged) + mu_near + Az),color = 'C4',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-kco_poly(full_rangec) - np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_near)**2 + kerr_poly(kco_poly(full_rangec) + mu_near)**2),zco_poly(full_rangec) + mu_near + zerr_poly(zco_poly(full_rangec) + mu_near),color = 'C5',label='5 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-kco_poly(full_rangec) + Azk + np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_near + Az)**2 + kerr_poly(kco_poly(full_rangec) + mu_near + Ak)**2), zco_poly(full_rangec) + mu_near + Az - zerr_poly(zco_poly(full_rangec) + mu_near + Az),color = 'C5',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-kdo_poly(full_ranged) - np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_far)**2 + kerr_poly(kdo_poly(full_ranged) + mu_far)**2),zdo_poly(full_ranged) + mu_far + zerr_poly(zdo_poly(full_ranged) + mu_far),color = 'C6',label='5 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(zdo_poly(full_ranged)-kdo_poly(full_ranged) + Azk + np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_far + Az)**2 + kerr_poly(kdo_poly(full_ranged) + mu_far + Ak)**2), zdo_poly(full_ranged) + mu_far + Az - zerr_poly(zdo_poly(full_ranged) + mu_far + Az),color = 'C6',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-kco_poly(full_rangec) - np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_far)**2 + kerr_poly(kco_poly(full_rangec) + mu_far)**2),zco_poly(full_rangec) + mu_far + zerr_poly(zco_poly(full_rangec) + mu_far),color = 'C8',label='5 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-kco_poly(full_rangec) + Azk + np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_far + Az)**2 + kerr_poly(kco_poly(full_rangec) + mu_far + Ak)**2), zco_poly(full_rangec) + mu_far + Az - zerr_poly(zco_poly(full_rangec) + mu_far + Az),color = 'C8',linewidth=4.0)\n",
    "plt.arrow(3.5,18,Azk,Az,width=0.01,length_includes_head=True,color='r')\n",
    "plt.text(3.5,17.9,'$A_v$ = 1.1',fontsize=30)\n",
    "plt.ylim(13,24)\n",
    "plt.xlim(.5,5)\n",
    "#plt.ylim(18.75,21)\n",
    "#plt.xlim(2,2.5)\n",
    "plt.legend(fontsize=32)\n",
    "plt.gca().invert_yaxis()\n",
    "plt.xlabel('Z-Ks Mag', fontsize=40)\n",
    "plt.ylabel('Z Mag', fontsize=40)\n",
    "plt.savefig('cmd_zk.png')\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "plt.figure(figsize=(40,20))\n",
    "plt.rc('xtick',labelsize=28)\n",
    "plt.rc('ytick',labelsize=28)\n",
    "plt.plot(mag_y-mag_j,mag_y,'.',color='grey')\n",
    "plt.plot(yd_poly(full_ranged)-jd_poly(full_ranged) - np.sqrt(yerr_poly(yd_poly(full_ranged) + mu_near)**2 + jerr_poly(jd_poly(full_ranged) + mu_near)**2),yd_poly(full_ranged) + mu_near + yerr_poly(yd_poly(full_ranged) + mu_near),color = 'C0',label='1 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(yd_poly(full_ranged)-jd_poly(full_ranged) + Ayj + np.sqrt(yerr_poly(yd_poly(full_ranged) + mu_near + Ay)**2 + jerr_poly(jd_poly(full_ranged) + mu_near + Aj)**2), yd_poly(full_ranged) + mu_near + Ay - yerr_poly(yd_poly(full_ranged) + mu_near + Ay),color = 'C0',linewidth=4.0)\n",
    "plt.plot(yc_poly(full_rangec)-jc_poly(full_rangec) - np.sqrt(yerr_poly(yc_poly(full_rangec) + mu_near)**2 + jerr_poly(jc_poly(full_rangec) + mu_near)**2),yc_poly(full_rangec) + mu_near + yerr_poly(yc_poly(full_rangec) + mu_near),color = 'C1',label='1 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(yc_poly(full_rangec)-jc_poly(full_rangec) + Ayj + np.sqrt(yerr_poly(yc_poly(full_rangec) + mu_near + Ay)**2 + jerr_poly(jc_poly(full_rangec) + mu_near + Aj)**2) , yc_poly(full_rangec) + mu_near + Ay - yerr_poly(yc_poly(full_rangec) + mu_near + Ay),color = 'C1',linewidth=4.0)\n",
    "plt.plot(yd_poly(full_ranged)-jd_poly(full_ranged) - np.sqrt(yerr_poly(yd_poly(full_ranged) + mu_far)**2 + jerr_poly(jd_poly(full_ranged) + mu_far)**2),yd_poly(full_ranged) + mu_far + yerr_poly(yd_poly(full_ranged) + mu_far),color = 'C2',label='1 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(yd_poly(full_ranged)-jd_poly(full_ranged) + Ayj + np.sqrt(yerr_poly(yd_poly(full_ranged) + mu_far + Ay)**2 + jerr_poly(jd_poly(full_ranged) + mu_far + Aj)**2), yd_poly(full_ranged) + mu_far + Ay - yerr_poly(yd_poly(full_ranged) + mu_far + Ay),color = 'C2',linewidth=4.0)\n",
    "plt.plot(yc_poly(full_rangec)-jc_poly(full_rangec) - np.sqrt(yerr_poly(yc_poly(full_rangec) + mu_far)**2 + jerr_poly(jc_poly(full_rangec) + mu_far)**2),yc_poly(full_rangec) + mu_far + yerr_poly(yc_poly(full_rangec) + mu_far),color = 'C3',label='1 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(yc_poly(full_rangec)-jc_poly(full_rangec) + Ayj + np.sqrt(yerr_poly(yc_poly(full_rangec) + mu_far + Ay)**2 + jerr_poly(jc_poly(full_rangec) + mu_far + Aj)**2), yc_poly(full_rangec) + mu_far + Ay - yerr_poly(yc_poly(full_rangec) + mu_far + Ay),color = 'C3',linewidth=4.0)\n",
    "plt.plot(ydo_poly(full_ranged)-jdo_poly(full_ranged) - np.sqrt(yerr_poly(ydo_poly(full_ranged) + mu_near)**2 + jerr_poly(jdo_poly(full_ranged) + mu_near)**2),ydo_poly(full_ranged) + mu_near + yerr_poly(ydo_poly(full_ranged) + mu_near),color = 'C4',label='5 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(ydo_poly(full_ranged)-jdo_poly(full_ranged) + Ayj + np.sqrt(yerr_poly(ydo_poly(full_ranged) + mu_near + Ay)**2 + jerr_poly(jdo_poly(full_ranged) + mu_near + Aj)**2), ydo_poly(full_ranged) + mu_near + Ay - yerr_poly(ydo_poly(full_ranged) + mu_near + Ay),color = 'C4',linewidth=4.0)\n",
    "plt.plot(yco_poly(full_rangec)-jco_poly(full_rangec) - np.sqrt(yerr_poly(yco_poly(full_rangec) + mu_near)**2 + jerr_poly(jco_poly(full_rangec) + mu_near)**2),yco_poly(full_rangec) + mu_near + yerr_poly(yco_poly(full_rangec) + mu_near),color = 'C5',label='5 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(yco_poly(full_rangec)-jco_poly(full_rangec) + Ayj + np.sqrt(yerr_poly(yco_poly(full_rangec) + mu_near + Ay)**2 + jerr_poly(jco_poly(full_rangec) + mu_near + Aj)**2), yco_poly(full_rangec) + mu_near + Ay - yerr_poly(yco_poly(full_rangec) + mu_near + Ay),color = 'C5',linewidth=4.0)\n",
    "plt.plot(ydo_poly(full_ranged)-jdo_poly(full_ranged) - np.sqrt(yerr_poly(ydo_poly(full_ranged) + mu_far)**2 + jerr_poly(jdo_poly(full_ranged) + mu_far)**2),ydo_poly(full_ranged) + mu_far + yerr_poly(ydo_poly(full_ranged) + mu_far),color = 'C6',label='5 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(ydo_poly(full_ranged)-jdo_poly(full_ranged) + Ayj + np.sqrt(yerr_poly(ydo_poly(full_ranged) + mu_far + Ay)**2 + jerr_poly(jdo_poly(full_ranged) + mu_far + Aj)**2), ydo_poly(full_ranged) + mu_far + Ay - yerr_poly(ydo_poly(full_ranged) + mu_far + Ay),color = 'C6',linewidth=4.0)\n",
    "plt.plot(yco_poly(full_rangec)-jco_poly(full_rangec) - np.sqrt(yerr_poly(yco_poly(full_rangec) + mu_far)**2 + jerr_poly(jco_poly(full_rangec) + mu_far)**2),yco_poly(full_rangec) + mu_far + yerr_poly(yco_poly(full_rangec) + mu_far),color = 'C8',label='5 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(yco_poly(full_rangec)-jco_poly(full_rangec) + Ayj + np.sqrt(yerr_poly(yco_poly(full_rangec) + mu_far + Ay)**2 + jerr_poly(jco_poly(full_rangec) + mu_far + Aj)**2), yco_poly(full_rangec) + mu_far + Ay - yerr_poly(yco_poly(full_rangec) + mu_far + Ay),color = 'C8',linewidth=4.0)\n",
    "plt.arrow(1.2,17,Ayj,Ay,width=0.01,length_includes_head=True,color='r')\n",
    "plt.text(1.2,16.9,'$A_v$ = 1.1',fontsize=30)\n",
    "plt.ylim(13,22.25)\n",
    "plt.xlim(0,2)\n",
    "#plt.ylim(18.75,21)\n",
    "#plt.xlim(2,2.5)\n",
    "plt.legend(fontsize=32)\n",
    "plt.gca().invert_yaxis()\n",
    "plt.xlabel('Y-J Mag', fontsize=40)\n",
    "plt.ylabel('Y Mag', fontsize=40)\n",
    "plt.savefig('cmd_yj.png')\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "plt.figure(figsize=(40,20))\n",
    "plt.rc('xtick',labelsize=28)\n",
    "plt.rc('ytick',labelsize=28)\n",
    "plt.plot(mag_y-mag_h,mag_y,'.',color='grey')\n",
    "plt.plot(yd_poly(full_ranged)-hd_poly(full_ranged) - np.sqrt(yerr_poly(yd_poly(full_ranged) + mu_near)**2 + herr_poly(hd_poly(full_ranged) + mu_near)**2),yd_poly(full_ranged) + mu_near + yerr_poly(yd_poly(full_ranged) + mu_near),color = 'C0',label='1 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(yd_poly(full_ranged)-hd_poly(full_ranged) + Ayh + np.sqrt(yerr_poly(yd_poly(full_ranged) + mu_near + Ay)**2 + herr_poly(hd_poly(full_ranged) + mu_near + Ah)**2), yd_poly(full_ranged) + mu_near + Ay - yerr_poly(yd_poly(full_ranged) + mu_near + Ay),color = 'C0',linewidth=4.0)\n",
    "plt.plot(yc_poly(full_rangec)-hc_poly(full_rangec) - np.sqrt(yerr_poly(yc_poly(full_rangec) + mu_near)**2 + herr_poly(hc_poly(full_rangec) + mu_near)**2),yc_poly(full_rangec) + mu_near + yerr_poly(yc_poly(full_rangec) + mu_near),color = 'C1',label='1 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(yc_poly(full_rangec)-hc_poly(full_rangec) + Ayh + np.sqrt(yerr_poly(yc_poly(full_rangec) + mu_near + Ay)**2 + herr_poly(hc_poly(full_rangec) + mu_near + Ah)**2) , yc_poly(full_rangec) + mu_near + Ay - yerr_poly(yc_poly(full_rangec) + mu_near + Ay),color = 'C1',linewidth=4.0)\n",
    "plt.plot(yd_poly(full_ranged)-hd_poly(full_ranged) - np.sqrt(yerr_poly(yd_poly(full_ranged) + mu_far)**2 + herr_poly(hd_poly(full_ranged) + mu_far)**2),yd_poly(full_ranged) + mu_far + yerr_poly(yd_poly(full_ranged) + mu_far),color = 'C2',label='1 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(yd_poly(full_ranged)-hd_poly(full_ranged) + Ayh + np.sqrt(yerr_poly(yd_poly(full_ranged) + mu_far + Ay)**2 + herr_poly(hd_poly(full_ranged) + mu_far + Ah)**2), yd_poly(full_ranged) + mu_far + Ay - yerr_poly(yd_poly(full_ranged) + mu_far + Ay),color = 'C2',linewidth=4.0)\n",
    "plt.plot(yc_poly(full_rangec)-hc_poly(full_rangec) - np.sqrt(yerr_poly(yc_poly(full_rangec) + mu_far)**2 + herr_poly(hc_poly(full_rangec) + mu_far)**2),yc_poly(full_rangec) + mu_far + yerr_poly(yc_poly(full_rangec) + mu_far),color = 'C3',label='1 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(yc_poly(full_rangec)-hc_poly(full_rangec) + Ayh + np.sqrt(yerr_poly(yc_poly(full_rangec) + mu_far + Ay)**2 + herr_poly(hc_poly(full_rangec) + mu_far + Ah)**2), yc_poly(full_rangec) + mu_far + Ay - yerr_poly(yc_poly(full_rangec) + mu_far + Ay),color = 'C3',linewidth=4.0)\n",
    "plt.plot(ydo_poly(full_ranged)-hdo_poly(full_ranged) - np.sqrt(yerr_poly(ydo_poly(full_ranged) + mu_near)**2 + herr_poly(hdo_poly(full_ranged) + mu_near)**2),ydo_poly(full_ranged) + mu_near + yerr_poly(ydo_poly(full_ranged) + mu_near),color = 'C4',label='5 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(ydo_poly(full_ranged)-hdo_poly(full_ranged) + Ayh + np.sqrt(yerr_poly(ydo_poly(full_ranged) + mu_near + Ay)**2 + herr_poly(hdo_poly(full_ranged) + mu_near + Ah)**2), ydo_poly(full_ranged) + mu_near + Ay - yerr_poly(ydo_poly(full_ranged) + mu_near + Ay),color = 'C4',linewidth=4.0)\n",
    "plt.plot(yco_poly(full_rangec)-hco_poly(full_rangec) - np.sqrt(yerr_poly(yco_poly(full_rangec) + mu_near)**2 + herr_poly(hco_poly(full_rangec) + mu_near)**2),yco_poly(full_rangec) + mu_near + yerr_poly(yco_poly(full_rangec) + mu_near),color = 'C5',label='5 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(yco_poly(full_rangec)-hco_poly(full_rangec) + Ayh + np.sqrt(yerr_poly(yco_poly(full_rangec) + mu_near + Ay)**2 + herr_poly(hco_poly(full_rangec) + mu_near + Ah)**2), yco_poly(full_rangec) + mu_near + Ay - yerr_poly(yco_poly(full_rangec) + mu_near + Ay),color = 'C5',linewidth=4.0)\n",
    "plt.plot(ydo_poly(full_ranged)-hdo_poly(full_ranged) - np.sqrt(yerr_poly(ydo_poly(full_ranged) + mu_far)**2 + herr_poly(hdo_poly(full_ranged) + mu_far)**2),ydo_poly(full_ranged) + mu_far + yerr_poly(ydo_poly(full_ranged) + mu_far),color = 'C6',label='5 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(ydo_poly(full_ranged)-hdo_poly(full_ranged) + Ayh + np.sqrt(yerr_poly(ydo_poly(full_ranged) + mu_far + Ay)**2 + herr_poly(hdo_poly(full_ranged) + mu_far + Ah)**2), ydo_poly(full_ranged) + mu_far + Ay - yerr_poly(ydo_poly(full_ranged) + mu_far + Ay),color = 'C6',linewidth=4.0)\n",
    "plt.plot(yco_poly(full_rangec)-hco_poly(full_rangec) - np.sqrt(yerr_poly(yco_poly(full_rangec) + mu_far)**2 + herr_poly(hco_poly(full_rangec) + mu_far)**2),yco_poly(full_rangec) + mu_far + yerr_poly(yco_poly(full_rangec) + mu_far),color = 'C8',label='5 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(yco_poly(full_rangec)-hco_poly(full_rangec) + Ayh + np.sqrt(yerr_poly(yco_poly(full_rangec) + mu_far + Ay)**2 + herr_poly(hco_poly(full_rangec) + mu_far + Ah)**2), yco_poly(full_rangec) + mu_far + Ay - yerr_poly(yco_poly(full_rangec) + mu_far + Ay),color = 'C8',linewidth=4.0)\n",
    "plt.arrow(1.75,17,Ayh,Ay,width=0.01,length_includes_head=True,color='r')\n",
    "plt.text(1.75,16.9,'$A_v$ = 1.1',fontsize=30)\n",
    "plt.ylim(13,22.25)\n",
    "plt.xlim(0.5,3)\n",
    "#plt.ylim(18.75,21)\n",
    "#plt.xlim(2,2.5)\n",
    "plt.legend(fontsize=32)\n",
    "plt.gca().invert_yaxis()\n",
    "plt.xlabel('Y-H Mag', fontsize=40)\n",
    "plt.ylabel('Y Mag', fontsize=40)\n",
    "plt.savefig('cmd_yh.png')\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "plt.figure(figsize=(40,20))\n",
    "plt.rc('xtick',labelsize=28)\n",
    "plt.rc('ytick',labelsize=28)\n",
    "plt.plot(mag_y-mag_k,mag_y,'.',color='grey')\n",
    "plt.plot(yd_poly(full_ranged)-kd_poly(full_ranged) - np.sqrt(yerr_poly(yd_poly(full_ranged) + mu_near)**2 + kerr_poly(kd_poly(full_ranged) + mu_near)**2),yd_poly(full_ranged) + mu_near + yerr_poly(yd_poly(full_ranged) + mu_near),color = 'C0',label='1 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(yd_poly(full_ranged)-kd_poly(full_ranged) + Ayk + np.sqrt(yerr_poly(yd_poly(full_ranged) + mu_near + Ay)**2 + kerr_poly(kd_poly(full_ranged) + mu_near + Ak)**2), yd_poly(full_ranged) + mu_near + Ay - yerr_poly(yd_poly(full_ranged) + mu_near + Ay),color = 'C0',linewidth=4.0)\n",
    "plt.plot(yc_poly(full_rangec)-kc_poly(full_rangec) - np.sqrt(yerr_poly(yc_poly(full_rangec) + mu_near)**2 + kerr_poly(kc_poly(full_rangec) + mu_near)**2),yc_poly(full_rangec) + mu_near + yerr_poly(yc_poly(full_rangec) + mu_near),color = 'C1',label='1 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(yc_poly(full_rangec)-kc_poly(full_rangec) + Ayk + np.sqrt(yerr_poly(yc_poly(full_rangec) + mu_near + Ay)**2 + kerr_poly(kc_poly(full_rangec) + mu_near + Ak)**2) , yc_poly(full_rangec) + mu_near + Ay - yerr_poly(yc_poly(full_rangec) + mu_near + Ay),color = 'C1',linewidth=4.0)\n",
    "plt.plot(yd_poly(full_ranged)-kd_poly(full_ranged) - np.sqrt(yerr_poly(yd_poly(full_ranged) + mu_far)**2 + kerr_poly(kd_poly(full_ranged) + mu_far)**2),yd_poly(full_ranged) + mu_far + yerr_poly(yd_poly(full_ranged) + mu_far),color = 'C2',label='1 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(yd_poly(full_ranged)-kd_poly(full_ranged) + Ayk + np.sqrt(yerr_poly(yd_poly(full_ranged) + mu_far + Ay)**2 + kerr_poly(kd_poly(full_ranged) + mu_far + Ak)**2), yd_poly(full_ranged) + mu_far + Ay - yerr_poly(yd_poly(full_ranged) + mu_far + Ay),color = 'C2',linewidth=4.0)\n",
    "plt.plot(yc_poly(full_rangec)-kc_poly(full_rangec) - np.sqrt(yerr_poly(yc_poly(full_rangec) + mu_far)**2 + kerr_poly(kc_poly(full_rangec) + mu_far)**2),yc_poly(full_rangec) + mu_far + yerr_poly(yc_poly(full_rangec) + mu_far),color = 'C3',label='1 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(yc_poly(full_rangec)-kc_poly(full_rangec) + Ayk + np.sqrt(yerr_poly(yc_poly(full_rangec) + mu_far + Ay)**2 + kerr_poly(kc_poly(full_rangec) + mu_far + Ak)**2), yc_poly(full_rangec) + mu_far + Ay - yerr_poly(yc_poly(full_rangec) + mu_far + Ay),color = 'C3',linewidth=4.0)\n",
    "plt.plot(ydo_poly(full_ranged)-kdo_poly(full_ranged) - np.sqrt(yerr_poly(ydo_poly(full_ranged) + mu_near)**2 + kerr_poly(kdo_poly(full_ranged) + mu_near)**2),ydo_poly(full_ranged) + mu_near + yerr_poly(ydo_poly(full_ranged) + mu_near),color = 'C4',label='5 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(ydo_poly(full_ranged)-kdo_poly(full_ranged) + Ayk + np.sqrt(yerr_poly(ydo_poly(full_ranged) + mu_near + Ay)**2 + kerr_poly(kdo_poly(full_ranged) + mu_near + Ak)**2), ydo_poly(full_ranged) + mu_near + Ay - yerr_poly(ydo_poly(full_ranged) + mu_near + Ay),color = 'C4',linewidth=4.0)\n",
    "plt.plot(yco_poly(full_rangec)-kco_poly(full_rangec) - np.sqrt(yerr_poly(yco_poly(full_rangec) + mu_near)**2 + kerr_poly(kco_poly(full_rangec) + mu_near)**2),yco_poly(full_rangec) + mu_near + yerr_poly(yco_poly(full_rangec) + mu_near),color = 'C5',label='5 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(yco_poly(full_rangec)-kco_poly(full_rangec) + Ayk + np.sqrt(yerr_poly(yco_poly(full_rangec) + mu_near + Ay)**2 + kerr_poly(kco_poly(full_rangec) + mu_near + Ak)**2), yco_poly(full_rangec) + mu_near + Ay - yerr_poly(yco_poly(full_rangec) + mu_near + Ay),color = 'C5',linewidth=4.0)\n",
    "plt.plot(ydo_poly(full_ranged)-kdo_poly(full_ranged) - np.sqrt(yerr_poly(ydo_poly(full_ranged) + mu_far)**2 + kerr_poly(kdo_poly(full_ranged) + mu_far)**2),ydo_poly(full_ranged) + mu_far + yerr_poly(ydo_poly(full_ranged) + mu_far),color = 'C6',label='5 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(ydo_poly(full_ranged)-kdo_poly(full_ranged) + Ayk + np.sqrt(yerr_poly(ydo_poly(full_ranged) + mu_far + Ay)**2 + kerr_poly(kdo_poly(full_ranged) + mu_far + Ak)**2), ydo_poly(full_ranged) + mu_far + Ay - yerr_poly(ydo_poly(full_ranged) + mu_far + Ay),color = 'C6',linewidth=4.0)\n",
    "plt.plot(yco_poly(full_rangec)-kco_poly(full_rangec) - np.sqrt(yerr_poly(yco_poly(full_rangec) + mu_far)**2 + kerr_poly(kco_poly(full_rangec) + mu_far)**2),yco_poly(full_rangec) + mu_far + yerr_poly(yco_poly(full_rangec) + mu_far),color = 'C8',label='5 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(yco_poly(full_rangec)-kco_poly(full_rangec) + Ayk + np.sqrt(yerr_poly(yco_poly(full_rangec) + mu_far + Ay)**2 + kerr_poly(kco_poly(full_rangec) + mu_far + Ak)**2), yco_poly(full_rangec) + mu_far + Ay - yerr_poly(yco_poly(full_rangec) + mu_far + Ay),color = 'C8',linewidth=4.0)\n",
    "plt.arrow(2,16,Ayk,Ay,width=0.01,length_includes_head=True,color='r')\n",
    "plt.text(2,15.9,'$A_v$ = 1.1',fontsize=30)\n",
    "plt.ylim(13,22.25)\n",
    "plt.xlim(0.75,4)\n",
    "#plt.ylim(18.75,21)\n",
    "#plt.xlim(2,2.5)\n",
    "plt.legend(fontsize=32)\n",
    "plt.gca().invert_yaxis()\n",
    "plt.xlabel('Y-Ks Mag', fontsize=40)\n",
    "plt.ylabel('Y Mag', fontsize=40)\n",
    "plt.savefig('cmd_yk.png')\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "plt.figure(figsize=(40,20))\n",
    "plt.rc('xtick',labelsize=28)\n",
    "plt.rc('ytick',labelsize=28)\n",
    "plt.plot(mag_j-mag_h,mag_j,'.',color='grey')\n",
    "plt.plot(jd_poly(full_ranged)-hd_poly(full_ranged) - np.sqrt(jerr_poly(jd_poly(full_ranged) + mu_near)**2 + herr_poly(hd_poly(full_ranged) + mu_near)**2), jd_poly(full_ranged) + mu_near + jerr_poly(jd_poly(full_ranged) + mu_near),color = 'C0',label='1 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(jd_poly(full_ranged)-hd_poly(full_ranged) + Ajh + np.sqrt(jerr_poly(jd_poly(full_ranged) + mu_near + Aj)**2 + herr_poly(hd_poly(full_ranged) + mu_near + Ah)**2), jd_poly(full_ranged) + mu_near + Aj - jerr_poly(jd_poly(full_ranged) + mu_near + Aj),color = 'C0',linewidth=4.0)\n",
    "plt.plot(jc_poly(full_rangec)-hc_poly(full_rangec) - np.sqrt(jerr_poly(jc_poly(full_rangec) + mu_near)**2 + herr_poly(hc_poly(full_rangec) + mu_near)**2), jc_poly(full_rangec) + mu_near + jerr_poly(jc_poly(full_rangec) + mu_near),color = 'C1',label='1 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(jc_poly(full_rangec)-hc_poly(full_rangec) + Ajh + np.sqrt(jerr_poly(jc_poly(full_rangec) + mu_near + Aj)**2 + herr_poly(hc_poly(full_rangec) + mu_near + Ah)**2) , jc_poly(full_rangec) + mu_near + Aj - jerr_poly(jc_poly(full_rangec) + mu_near + Aj),color = 'C1',linewidth=4.0)\n",
    "plt.plot(jd_poly(full_ranged)-hd_poly(full_ranged) - np.sqrt(jerr_poly(jd_poly(full_ranged) + mu_far)**2 + herr_poly(hd_poly(full_ranged) + mu_far)**2), jd_poly(full_ranged) + mu_far + jerr_poly(jd_poly(full_ranged) + mu_far),color = 'C2',label='1 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(jd_poly(full_ranged)-hd_poly(full_ranged) + Ajh + np.sqrt(jerr_poly(jd_poly(full_ranged) + mu_far + Aj)**2 + herr_poly(hd_poly(full_ranged) + mu_far + Ah)**2), jd_poly(full_ranged) + mu_far + Aj - jerr_poly(jd_poly(full_ranged) + mu_far + Aj),color = 'C2',linewidth=4.0)\n",
    "plt.plot(jc_poly(full_rangec)-hc_poly(full_rangec) - np.sqrt(jerr_poly(jc_poly(full_rangec) + mu_far)**2 + herr_poly(hc_poly(full_rangec) + mu_far)**2), jc_poly(full_rangec) + mu_far + jerr_poly(jc_poly(full_rangec) + mu_far),color = 'C3',label='1 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(jc_poly(full_rangec)-hc_poly(full_rangec) + Ajh + np.sqrt(jerr_poly(jc_poly(full_rangec) + mu_far + Aj)**2 + herr_poly(hc_poly(full_rangec) + mu_far + Ah)**2), jc_poly(full_rangec) + mu_far + Aj - jerr_poly(jc_poly(full_rangec) + mu_far + Aj),color = 'C3',linewidth=4.0)\n",
    "plt.plot(jdo_poly(full_ranged)-hdo_poly(full_ranged) - np.sqrt(jerr_poly(jdo_poly(full_ranged) + mu_near)**2 + herr_poly(hdo_poly(full_ranged) + mu_near)**2), jdo_poly(full_ranged) + mu_near + jerr_poly(jdo_poly(full_ranged) + mu_near),color = 'C4',label='5 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(jdo_poly(full_ranged)-hdo_poly(full_ranged) + Ajh + np.sqrt(jerr_poly(jdo_poly(full_ranged) + mu_near + Aj)**2 + herr_poly(hdo_poly(full_ranged) + mu_near + Ah)**2), jdo_poly(full_ranged) + mu_near + Aj - jerr_poly(jdo_poly(full_ranged) + mu_near + Aj),color = 'C4',linewidth=4.0)\n",
    "plt.plot(jco_poly(full_rangec)-hco_poly(full_rangec) - np.sqrt(jerr_poly(jco_poly(full_rangec) + mu_near)**2 + herr_poly(hco_poly(full_rangec) + mu_near)**2), jco_poly(full_rangec) + mu_near + jerr_poly(jco_poly(full_rangec) + mu_near),color = 'C5',label='5 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(jco_poly(full_rangec)-hco_poly(full_rangec) + Ajh + np.sqrt(jerr_poly(jco_poly(full_rangec) + mu_near + Aj)**2 + herr_poly(hco_poly(full_rangec) + mu_near + Ah)**2), jco_poly(full_rangec) + mu_near + Aj - jerr_poly(jco_poly(full_rangec) + mu_near + Aj),color = 'C5',linewidth=4.0)\n",
    "plt.plot(jdo_poly(full_ranged)-hdo_poly(full_ranged) - np.sqrt(jerr_poly(jdo_poly(full_ranged) + mu_far)**2 + herr_poly(hdo_poly(full_ranged) + mu_far)**2), jdo_poly(full_ranged) + mu_far + jerr_poly(jdo_poly(full_ranged) + mu_far),color = 'C6',label='5 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(jdo_poly(full_ranged)-hdo_poly(full_ranged) + Ajh + np.sqrt(jerr_poly(jdo_poly(full_ranged) + mu_far + Aj)**2 + herr_poly(hdo_poly(full_ranged) + mu_far + Ah)**2), jdo_poly(full_ranged) + mu_far + Aj - jerr_poly(jdo_poly(full_ranged) + mu_far + Aj),color = 'C6',linewidth=4.0)\n",
    "plt.plot(jco_poly(full_rangec)-hco_poly(full_rangec) - np.sqrt(jerr_poly(jco_poly(full_rangec) + mu_far)**2 + herr_poly(hco_poly(full_rangec) + mu_far)**2), jco_poly(full_rangec) + mu_far + jerr_poly(jco_poly(full_rangec) + mu_far),color = 'C8',label='5 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(jco_poly(full_rangec)-hco_poly(full_rangec) + Ajh + np.sqrt(jerr_poly(jco_poly(full_rangec) + mu_far + Aj)**2 + herr_poly(hco_poly(full_rangec) + mu_far + Ah)**2), jco_poly(full_rangec) + mu_far + Aj - jerr_poly(jco_poly(full_rangec) + mu_far + Aj),color = 'C8',linewidth=4.0)\n",
    "plt.arrow(1,16,Ajh,Aj,width=0.01,length_includes_head=True,color='r')\n",
    "plt.text(1,15.9,'$A_v$ = 1.1',fontsize=30)\n",
    "plt.ylim(12.5,21.5)\n",
    "plt.xlim(-0.5,2)\n",
    "#plt.ylim(18.75,21)\n",
    "#plt.xlim(2,2.5)\n",
    "plt.legend(fontsize=32)\n",
    "plt.gca().invert_yaxis()\n",
    "plt.xlabel('J-H Mag', fontsize=40)\n",
    "plt.ylabel('J Mag', fontsize=40)\n",
    "plt.savefig('cmd_jh.png')\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "plt.figure(figsize=(40,20))\n",
    "plt.rc('xtick',labelsize=28)\n",
    "plt.rc('ytick',labelsize=28)\n",
    "plt.plot(mag_j-mag_k,mag_j,'.',color='grey')\n",
    "plt.plot(jd_poly(full_ranged)-kd_poly(full_ranged) - np.sqrt(jerr_poly(jd_poly(full_ranged) + mu_near)**2 + kerr_poly(kd_poly(full_ranged) + mu_near)**2),jd_poly(full_ranged) + mu_near + jerr_poly(jd_poly(full_ranged) + mu_near),color = 'C0',label='1 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(jd_poly(full_ranged)-kd_poly(full_ranged) + Ajk + np.sqrt(jerr_poly(jd_poly(full_ranged) + mu_near + Aj)**2 + kerr_poly(kd_poly(full_ranged) + mu_near + Ak)**2), jd_poly(full_ranged) + mu_near + Aj - jerr_poly(jd_poly(full_ranged) + mu_near + Aj),color = 'C0',linewidth=4.0)\n",
    "plt.plot(jc_poly(full_rangec)-kc_poly(full_rangec) - np.sqrt(jerr_poly(jc_poly(full_rangec) + mu_near)**2 + kerr_poly(kc_poly(full_rangec) + mu_near)**2),jc_poly(full_rangec) + mu_near + jerr_poly(jc_poly(full_rangec) + mu_near),color = 'C1',label='1 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(jc_poly(full_rangec)-kc_poly(full_rangec) + Ajk + np.sqrt(jerr_poly(jc_poly(full_rangec) + mu_near + Aj)**2 + kerr_poly(kc_poly(full_rangec) + mu_near + Ak)**2) , jc_poly(full_rangec) + mu_near + Aj - jerr_poly(jc_poly(full_rangec) + mu_near + Aj),color = 'C1',linewidth=4.0)\n",
    "plt.plot(jd_poly(full_ranged)-kd_poly(full_ranged) - np.sqrt(jerr_poly(jd_poly(full_ranged) + mu_far)**2 + kerr_poly(kd_poly(full_ranged) + mu_far)**2),jd_poly(full_ranged) + mu_far + jerr_poly(jd_poly(full_ranged) + mu_far),color = 'C2',label='1 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(jd_poly(full_ranged)-kd_poly(full_ranged) + Ajk + np.sqrt(jerr_poly(jd_poly(full_ranged) + mu_far + Aj)**2 + kerr_poly(kd_poly(full_ranged) + mu_far + Ak)**2), jd_poly(full_ranged) + mu_far + Aj - jerr_poly(jd_poly(full_ranged) + mu_far + Aj),color = 'C2',linewidth=4.0)\n",
    "plt.plot(jc_poly(full_rangec)-kc_poly(full_rangec) - np.sqrt(jerr_poly(jc_poly(full_rangec) + mu_far)**2 + kerr_poly(kc_poly(full_rangec) + mu_far)**2),jc_poly(full_rangec) + mu_far + jerr_poly(jc_poly(full_rangec) + mu_far),color = 'C3',label='1 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(jc_poly(full_rangec)-kc_poly(full_rangec) + Ajk + np.sqrt(jerr_poly(jc_poly(full_rangec) + mu_far + Aj)**2 + kerr_poly(kc_poly(full_rangec) + mu_far + Ak)**2), jc_poly(full_rangec) + mu_far + Aj - jerr_poly(jc_poly(full_rangec) + mu_far + Aj),color = 'C3',linewidth=4.0)\n",
    "plt.plot(jdo_poly(full_ranged)-kdo_poly(full_ranged) - np.sqrt(jerr_poly(jdo_poly(full_ranged) + mu_near)**2 + kerr_poly(kdo_poly(full_ranged) + mu_near)**2),jdo_poly(full_ranged) + mu_near + jerr_poly(jdo_poly(full_ranged) + mu_near),color = 'C4',label='5 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(jdo_poly(full_ranged)-kdo_poly(full_ranged) + Ajk + np.sqrt(jerr_poly(jdo_poly(full_ranged) + mu_near + Aj)**2 + kerr_poly(kdo_poly(full_ranged) + mu_near + Ak)**2), jdo_poly(full_ranged) + mu_near + Aj - jerr_poly(jdo_poly(full_ranged) + mu_near + Aj),color = 'C4',linewidth=4.0)\n",
    "plt.plot(jco_poly(full_rangec)-kco_poly(full_rangec) - np.sqrt(jerr_poly(jco_poly(full_rangec) + mu_near)**2 + kerr_poly(kco_poly(full_rangec) + mu_near)**2),jco_poly(full_rangec) + mu_near + jerr_poly(jco_poly(full_rangec) + mu_near),color = 'C5',label='5 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(jco_poly(full_rangec)-kco_poly(full_rangec) + Ajk + np.sqrt(jerr_poly(jco_poly(full_rangec) + mu_near + Aj)**2 + kerr_poly(kco_poly(full_rangec) + mu_near + Ak)**2), jco_poly(full_rangec) + mu_near + Aj - jerr_poly(jco_poly(full_rangec) + mu_near + Aj),color = 'C5',linewidth=4.0)\n",
    "plt.plot(jdo_poly(full_ranged)-kdo_poly(full_ranged) - np.sqrt(jerr_poly(jdo_poly(full_ranged) + mu_far)**2 + kerr_poly(kdo_poly(full_ranged) + mu_far)**2),jdo_poly(full_ranged) + mu_far + jerr_poly(jdo_poly(full_ranged) + mu_far),color = 'C6',label='5 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(jdo_poly(full_ranged)-kdo_poly(full_ranged) + Ajk + np.sqrt(jerr_poly(jdo_poly(full_ranged) + mu_far + Aj)**2 + kerr_poly(kdo_poly(full_ranged) + mu_far + Ak)**2), jdo_poly(full_ranged) + mu_far + Aj - jerr_poly(jdo_poly(full_ranged) + mu_far + Aj),color = 'C6',linewidth=4.0)\n",
    "plt.plot(jco_poly(full_rangec)-kco_poly(full_rangec) - np.sqrt(jerr_poly(jco_poly(full_rangec) + mu_far)**2 + kerr_poly(kco_poly(full_rangec) + mu_far)**2),jco_poly(full_rangec) + mu_far + jerr_poly(jco_poly(full_rangec) + mu_far),color = 'C8',label='5 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(jco_poly(full_rangec)-kco_poly(full_rangec) + Ajk + np.sqrt(jerr_poly(jco_poly(full_rangec) + mu_far + Aj)**2 + kerr_poly(kco_poly(full_rangec) + mu_far + Ak)**2), jco_poly(full_rangec) + mu_far + Aj - jerr_poly(jco_poly(full_rangec) + mu_far + Aj),color = 'C8',linewidth=4.0)\n",
    "plt.arrow(1.25,14,Ajk,Aj,width=0.01,length_includes_head=True,color='r')\n",
    "plt.text(1.25,13.9,'$A_v$ = 1.1',fontsize=30)\n",
    "plt.ylim(12.5,21.5)\n",
    "plt.xlim(0.25,2.5)\n",
    "#plt.ylim(18.75,21)\n",
    "#plt.xlim(2,2.5)\n",
    "plt.legend(fontsize=32)\n",
    "plt.gca().invert_yaxis()\n",
    "plt.xlabel('J-Ks Mag', fontsize=40)\n",
    "plt.ylabel('J Mag', fontsize=40)\n",
    "plt.savefig('cmd_jk.png')\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "plt.figure(figsize=(40,20))\n",
    "plt.rc('xtick',labelsize=28)\n",
    "plt.rc('ytick',labelsize=28)\n",
    "plt.plot(mag_h-mag_k,mag_h,'.',color='grey')\n",
    "plt.plot(hd_poly(full_ranged)-kd_poly(full_ranged) - np.sqrt(herr_poly(hd_poly(full_ranged) + mu_near)**2 + kerr_poly(kd_poly(full_ranged) + mu_near)**2),hd_poly(full_ranged) + mu_near + herr_poly(hd_poly(full_ranged) + mu_near),color = 'C0',label='1 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(hd_poly(full_ranged)-kd_poly(full_ranged) + Ahk + np.sqrt(herr_poly(hd_poly(full_ranged) + mu_near + Ah)**2 + kerr_poly(kd_poly(full_ranged) + mu_near + Ak)**2), hd_poly(full_ranged) + mu_near + Ah - herr_poly(hd_poly(full_ranged) + mu_near + Ah),color = 'C0',linewidth=4.0)\n",
    "plt.plot(hc_poly(full_rangec)-kc_poly(full_rangec) - np.sqrt(herr_poly(hc_poly(full_rangec) + mu_near)**2 + kerr_poly(kc_poly(full_rangec) + mu_near)**2),hc_poly(full_rangec) + mu_near + herr_poly(hc_poly(full_rangec) + mu_near),color = 'C1',label='1 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(hc_poly(full_rangec)-kc_poly(full_rangec) + Ahk + np.sqrt(herr_poly(hc_poly(full_rangec) + mu_near + Ah)**2 + kerr_poly(kc_poly(full_rangec) + mu_near + Ak)**2) , hc_poly(full_rangec) + mu_near + Ah - herr_poly(hc_poly(full_rangec) + mu_near + Ah),color = 'C1',linewidth=4.0)\n",
    "plt.plot(hd_poly(full_ranged)-kd_poly(full_ranged) - np.sqrt(herr_poly(hd_poly(full_ranged) + mu_far)**2 + kerr_poly(kd_poly(full_ranged) + mu_far)**2),hd_poly(full_ranged) + mu_far + herr_poly(hd_poly(full_ranged) + mu_far),color = 'C2',label='1 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(hd_poly(full_ranged)-kd_poly(full_ranged) + Ahk + np.sqrt(herr_poly(hd_poly(full_ranged) + mu_far + Ah)**2 + kerr_poly(kd_poly(full_ranged) + mu_far + Ak)**2), hd_poly(full_ranged) + mu_far + Ah - herr_poly(hd_poly(full_ranged) + mu_far + Ah),color = 'C2',linewidth=4.0)\n",
    "plt.plot(hc_poly(full_rangec)-kc_poly(full_rangec) - np.sqrt(herr_poly(hc_poly(full_rangec) + mu_far)**2 + kerr_poly(kc_poly(full_rangec) + mu_far)**2),hc_poly(full_rangec) + mu_far + herr_poly(hc_poly(full_rangec) + mu_far),color = 'C3',label='1 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(hc_poly(full_rangec)-kc_poly(full_rangec) + Ahk + np.sqrt(herr_poly(hc_poly(full_rangec) + mu_far + Ah)**2 + kerr_poly(kc_poly(full_rangec) + mu_far + Ak)**2), hc_poly(full_rangec) + mu_far + Ah - herr_poly(hc_poly(full_rangec) + mu_far + Ah),color = 'C3',linewidth=4.0)\n",
    "plt.plot(hdo_poly(full_ranged)-kdo_poly(full_ranged) - np.sqrt(herr_poly(hdo_poly(full_ranged) + mu_near)**2 + kerr_poly(kdo_poly(full_ranged) + mu_near)**2),hdo_poly(full_ranged) + mu_near + herr_poly(hdo_poly(full_ranged) + mu_near),color = 'C4',label='5 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(hdo_poly(full_ranged)-kdo_poly(full_ranged) + Ahk + np.sqrt(herr_poly(hdo_poly(full_ranged) + mu_near + Ah)**2 + kerr_poly(kdo_poly(full_ranged) + mu_near + Ak)**2), hdo_poly(full_ranged) + mu_near + Ah - herr_poly(hdo_poly(full_ranged) + mu_near + Ah),color = 'C4',linewidth=4.0)\n",
    "plt.plot(hco_poly(full_rangec)-kco_poly(full_rangec) - np.sqrt(herr_poly(hco_poly(full_rangec) + mu_near)**2 + kerr_poly(kco_poly(full_rangec) + mu_near)**2),hco_poly(full_rangec) + mu_near + herr_poly(hco_poly(full_rangec) + mu_near),color = 'C5',label='5 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "plt.plot(hco_poly(full_rangec)-kco_poly(full_rangec) + Ahk + np.sqrt(herr_poly(hco_poly(full_rangec) + mu_near + Ah)**2 + kerr_poly(kco_poly(full_rangec) + mu_near + Ak)**2), hco_poly(full_rangec) + mu_near + Ah - herr_poly(hco_poly(full_rangec) + mu_near + Ah),color = 'C5',linewidth=4.0)\n",
    "plt.plot(hdo_poly(full_ranged)-kdo_poly(full_ranged) - np.sqrt(herr_poly(hdo_poly(full_ranged) + mu_far)**2 + kerr_poly(kdo_poly(full_ranged) + mu_far)**2),hdo_poly(full_ranged) + mu_far + herr_poly(hdo_poly(full_ranged) + mu_far),color = 'C6',label='5 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "plt.plot(hdo_poly(full_ranged)-kdo_poly(full_ranged) + Ahk + np.sqrt(herr_poly(hdo_poly(full_ranged) + mu_far + Ah)**2 + kerr_poly(kdo_poly(full_ranged) + mu_far + Ak)**2), hdo_poly(full_ranged) + mu_far + Ah - herr_poly(hdo_poly(full_ranged) + mu_far + Ah),color = 'C6',linewidth=4.0)\n",
    "plt.plot(hco_poly(full_rangec)-kco_poly(full_rangec) - np.sqrt(herr_poly(hco_poly(full_rangec) + mu_far)**2 + kerr_poly(kco_poly(full_rangec) + mu_far)**2),hco_poly(full_rangec) + mu_far + herr_poly(hco_poly(full_rangec) + mu_far),color = 'C8',label='5 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "plt.plot(hco_poly(full_rangec)-kco_poly(full_rangec) + Ahk + np.sqrt(herr_poly(hco_poly(full_rangec) + mu_far + Ah)**2 + kerr_poly(kco_poly(full_rangec) + mu_far + Ak)**2), hco_poly(full_rangec) + mu_far + Ah - herr_poly(hco_poly(full_rangec) + mu_far + Ah),color = 'C8',linewidth=4.0)\n",
    "plt.arrow(0.7,14,Ahk,Ah,width=0.01,length_includes_head=True,color='r')\n",
    "plt.text(0.7,13.9,'$A_v$ = 1.1',fontsize=30)\n",
    "plt.ylim(12,20)\n",
    "plt.xlim(-0.25,1.5)\n",
    "#plt.ylim(18.75,21)\n",
    "#plt.xlim(2,2.5)\n",
    "plt.legend(fontsize=32)\n",
    "plt.gca().invert_yaxis()\n",
    "plt.xlabel('H-Ks Mag', fontsize=40)\n",
    "plt.ylabel('H Mag', fontsize=40)\n",
    "plt.savefig('cmd_hk.png')\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "plt.figure(figsize=(40,20))\n",
    "plt.rc('xtick',labelsize=28)\n",
    "plt.rc('ytick',labelsize=28)\n",
    "plt.plot(mag_z-mag_j,mag_z,'.',color='grey')\n",
    "#plt.plot(zd_poly(full_ranged)-jd_poly(full_ranged) - np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_near)**2 + jerr_poly(jd_poly(full_ranged) + mu_near)**2),zd_poly(full_ranged) + mu_near + zerr_poly(zd_poly(full_ranged) + mu_near),color = 'C0',label='1 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "#plt.plot(zd_poly(full_ranged)-jd_poly(full_ranged) + Azj + np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_near + Az)**2 + jerr_poly(jd_poly(full_ranged) + mu_near + Aj)**2), zd_poly(full_ranged) + mu_near + Az - zerr_poly(zd_poly(full_ranged) + mu_near + Az),color = 'C0',linewidth=4.0)\n",
    "#plt.plot(zc_poly(full_rangec)-jc_poly(full_rangec) - np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_near)**2 + jerr_poly(jc_poly(full_rangec) + mu_near)**2),zc_poly(full_rangec) + mu_near + zerr_poly(zc_poly(full_rangec) + mu_near),color = 'C1',label='1 Mya - 350 pc - Cond',linewidth=4.0)\n",
    "#plt.plot(zc_poly(full_rangec)-jc_poly(full_rangec) + Azj + np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_near + Az)**2 + jerr_poly(jc_poly(full_rangec) + mu_near + Aj)**2) , zc_poly(full_rangec) + mu_near + Az - zerr_poly(zc_poly(full_rangec) + mu_near + Az),color = 'C1',linewidth=4.0)\n",
    "#plt.plot(zd_poly(full_ranged)-jd_poly(full_ranged) - np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_far)**2 + jerr_poly(jd_poly(full_ranged) + mu_far)**2),zd_poly(full_ranged) + mu_far + zerr_poly(zd_poly(full_ranged) + mu_far),color = 'C2',label='1 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "#plt.plot(zd_poly(full_ranged)-jd_poly(full_ranged) + Azj + np.sqrt(zerr_poly(zd_poly(full_ranged) + mu_far + Az)**2 + jerr_poly(jd_poly(full_ranged) + mu_far + Aj)**2), zd_poly(full_ranged) + mu_far + Az - zerr_poly(zd_poly(full_ranged) + mu_far + Az),color = 'C2',linewidth=4.0)\n",
    "#plt.plot(zc_poly(full_rangec)-jc_poly(full_rangec) - np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_far)**2 + jerr_poly(jc_poly(full_rangec) + mu_far)**2),zc_poly(full_rangec) + mu_far + zerr_poly(zc_poly(full_rangec) + mu_far),color = 'C3',label='1 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "#plt.plot(zc_poly(full_rangec)-jc_poly(full_rangec) + Azj + np.sqrt(zerr_poly(zc_poly(full_rangec) + mu_far + Az)**2 + jerr_poly(jc_poly(full_rangec) + mu_far + Aj)**2), zc_poly(full_rangec) + mu_far + Az - zerr_poly(zc_poly(full_rangec) + mu_far + Az),color = 'C3',linewidth=4.0)\n",
    "#plt.plot(zdo_poly(full_ranged)-jdo_poly(full_ranged) - np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_near)**2 + jerr_poly(jdo_poly(full_ranged) + mu_near)**2),zdo_poly(full_ranged) + mu_near + zerr_poly(zdo_poly(full_ranged) + mu_near),color = 'C4',label='5 Mya - 350 pc - Dusty',linewidth=4.0)\n",
    "#plt.plot(zdo_poly(full_ranged)-jdo_poly(full_ranged) + Azj + np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_near + Az)**2 + jerr_poly(jdo_poly(full_ranged) + mu_near + Aj)**2), zdo_poly(full_ranged) + mu_near + Az - zerr_poly(zdo_poly(full_ranged) + mu_near + Az),color = 'C4',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-jco_poly(full_rangec) - np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_near)**2 + jerr_poly(jco_poly(full_rangec) + mu_near)**2),zco_poly(full_rangec) + mu_near + zerr_poly(zco_poly(full_rangec) + mu_near),color = 'C0',label='5 Mya - 350 pc - Cond w/ error',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-jco_poly(full_rangec) + Azj + np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_near + Az)**2 + jerr_poly(jco_poly(full_rangec) + mu_near + Aj)**2), zco_poly(full_rangec) + mu_near + Az - zerr_poly(zco_poly(full_rangec) + mu_near + Az),color = 'C0',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-jco_poly(full_rangec) ,zco_poly(full_rangec) + mu_near , color = 'C1',label='5 Mya - 350 pc - Cond w/o error',linewidth=4.0)\n",
    "plt.plot(zco_poly(full_rangec)-jco_poly(full_rangec) + Azj , zco_poly(full_rangec) + mu_near + Az,color = 'C1',linewidth=4.0)\n",
    "#plt.plot(zdo_poly(full_ranged)-jdo_poly(full_ranged) - np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_far)**2 + jerr_poly(jdo_poly(full_ranged) + mu_far)**2),zdo_poly(full_ranged) + mu_far + zerr_poly(zdo_poly(full_ranged) + mu_far),color = 'C6',label='5 Mya - 420 pc - Dusty',linewidth=4.0)\n",
    "#plt.plot(zdo_poly(full_ranged)-jdo_poly(full_ranged) + Azj + np.sqrt(zerr_poly(zdo_poly(full_ranged) + mu_far + Az)**2 + jerr_poly(jdo_poly(full_ranged) + mu_far + Aj)**2), zdo_poly(full_ranged) + mu_far + Az - zerr_poly(zdo_poly(full_ranged) + mu_far + Az),color = 'C6',linewidth=4.0)\n",
    "#plt.plot(zco_poly(full_rangec)-jco_poly(full_rangec) - np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_far)**2 + jerr_poly(jco_poly(full_rangec) + mu_far)**2),zco_poly(full_rangec) + mu_far + zerr_poly(zco_poly(full_rangec) + mu_far),color = 'C8',label='5 Mya - 420 pc - Cond',linewidth=4.0)\n",
    "#plt.plot(zco_poly(full_rangec)-jco_poly(full_rangec) + Azj + np.sqrt(zerr_poly(zco_poly(full_rangec) + mu_far + Az)**2 + jerr_poly(jco_poly(full_rangec) + mu_far + Aj)**2), zco_poly(full_rangec) + mu_far + Az - zerr_poly(zco_poly(full_rangec) + mu_far + Az),color = 'C8',linewidth=4.0)\n",
    "plt.arrow(2.1,19.5,Azj,Az,width=0.01,length_includes_head=True,color='r')\n",
    "plt.text(2.1,19.4,'$A_v$ = 1.1',fontsize=30)\n",
    "plt.ylim(17.5,21.5)\n",
    "plt.xlim(1,3)\n",
    "#plt.ylim(18.75,21)\n",
    "#plt.xlim(2,2.5)\n",
    "plt.legend(fontsize=32)\n",
    "plt.gca().invert_yaxis()\n",
    "plt.xlabel('Z-J Mag', fontsize=40)\n",
    "plt.ylabel('Z Mag', fontsize=40)\n",
    "plt.savefig('skinny_part_woerror.png')\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def candidate_stats(file,output_name,start_temp=1200,max_temp=4500):\n",
    "    \n",
    "    '''This program finds the temperatures and masses of each object in the input file, and then creates\n",
    "    a table with that information. This version uses all possible set-ups. Following ones use specific \n",
    "    types of set-ups.'''\n",
    "\n",
    "    hdul = fits.open(file)\n",
    "    \n",
    "    RA_ICRS = hdul[1].data['RA_ICRS']\n",
    "    Dec_ICRS = hdul[1].data['Dec_ICRS']\n",
    "    zmag = hdul[1].data['Mag_Z']\n",
    "    ymag = hdul[1].data['Mag_Y']\n",
    "    jmag = hdul[1].data['Mag_J']\n",
    "    hmag = hdul[1].data['Mag_H']\n",
    "    kmag = hdul[1].data['Mag_K']\n",
    "\n",
    "    temp_min = []\n",
    "    temp_mean = []\n",
    "    temp_max = []\n",
    "    mass_min = []\n",
    "    mass_mean_y = []\n",
    "    mass_mean_o = []\n",
    "    mass_max = []\n",
    "\n",
    "    for y in zmag:\n",
    "        temps = []\n",
    "\n",
    "        x = start_temp\n",
    "        while (zd_poly(x) + mu_near + zerr_poly(zd_poly(x) + mu_near)) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zd_poly(x) + mu_near + Az - zerr_poly(zd_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_near + zerr_poly(zc_poly(x) + mu_near) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_near + Az - zerr_poly(zc_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zd_poly(x) + mu_far + zerr_poly(zd_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zd_poly(x) + mu_far + Az - zerr_poly(zd_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_far + zerr_poly(zc_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_far + Az - zerr_poly(zc_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_near + zerr_poly(zdo_poly(x) + mu_near) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_near + Az - zerr_poly(zdo_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_near + zerr_poly(zco_poly(x) + mu_near) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_near + Az - zerr_poly(zco_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_far + zerr_poly(zdo_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_far + Az - zerr_poly(zdo_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_far + zerr_poly(zco_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_far + Az - zerr_poly(zco_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        temp_min.append(np.min(temps))\n",
    "        temp_mean.append(np.mean(temps))\n",
    "        temp_max.append(np.max(temps))\n",
    "        if np.min([mass_poly(np.min(temps)),masso_poly(np.min(temps))]) > 0.0016:\n",
    "            mass_min.append(np.min([mass_poly(np.min(temps)),masso_poly(np.min(temps))]))\n",
    "        else:\n",
    "            mass_min.append(np.max([mass_poly(np.min(temps)),masso_poly(np.min(temps))]))\n",
    "        mass_mean_y.append(mass_poly(np.mean(temps)))\n",
    "        mass_mean_o.append(masso_poly(np.mean(temps)))\n",
    "        mass_max.append(np.max([mass_poly(np.max(temps)),masso_poly(np.max(temps))]))\n",
    "        \n",
    "    for x in range(len(temp_min)):\n",
    "        if mass_min[x] > mass_mean_y[x]:\n",
    "            mass_mean_y[x] = mass_min[x]\n",
    "        if mass_min[x] > mass_mean_o[x]:\n",
    "            mass_mean_o[x] = mass_min[x]\n",
    "        if mass_max[x] < mass_mean_y[x]:\n",
    "            mass_mean_y[x] = mass_max[x]\n",
    "        if mass_max[x] < mass_mean_o[x]:\n",
    "            mass_mean_o[x] = mass_max[x]\n",
    "    \n",
    "    col1 = fits.Column(name='RA_ICRS',format='D',unit='degree',array=RA_ICRS)\n",
    "    col2 = fits.Column(name='Dec_ICRS',format='D',unit='degree',array=Dec_ICRS)\n",
    "    col3 = fits.Column(name='Mag_Z',format='D',unit='mag',array=zmag)\n",
    "    col4 = fits.Column(name='Mag_Y',format='D',unit='mag',array=ymag)\n",
    "    col5 = fits.Column(name='Mag_J',format='D',unit='mag',array=jmag)\n",
    "    col6 = fits.Column(name='Mag_H',format='D',unit='mag',array=hmag)\n",
    "    col7 = fits.Column(name='Mag_K',format='D',unit='mag',array=kmag)\n",
    "    col8 = fits.Column(name='Min_Temp',format='F',unit='K',array=temp_min)\n",
    "    col9 = fits.Column(name='Mean_Temp',format='F',unit='K',array=temp_mean)\n",
    "    col10 = fits.Column(name='Max_Temp',format='F',unit='K',array=temp_max)\n",
    "    col11 = fits.Column(name='Min_Mass',format='F',unit='M_sol',array=mass_min)\n",
    "    col12 = fits.Column(name='Mean_Mass_Young',format='F',unit='M_sol',array=mass_mean_y)\n",
    "    col13 = fits.Column(name='Mean_Mass_Old',format='F',unit='M_sol',array=mass_mean_o)\n",
    "    col14 = fits.Column(name='Max_Mass',format='F',unit='M_sol',array=mass_max)\n",
    "    \n",
    "    cols = fits.ColDefs([col1,col2,col3,col4,col5,col6,col7,col8,col9,col10,col11,col12,col13,col14])\n",
    "    \n",
    "    '''Here we make the fits file'''\n",
    "\n",
    "    output_fits = output_name + '.fits'\n",
    "    data = fits.BinTableHDU.from_columns(cols)\n",
    "    data.writeto(output_fits)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def candidate_stats_young(file,output_name,start_temp=1200,max_temp=4500):\n",
    "    \n",
    "    '''This program finds the temperatures and masses of each object in the input file, and then creates\n",
    "    a table with that information. This version uses 1Myr set-ups.'''\n",
    "\n",
    "    hdul = fits.open(file)\n",
    "    \n",
    "    RA_ICRS = hdul[1].data['RA_ICRS']\n",
    "    Dec_ICRS = hdul[1].data['Dec_ICRS']\n",
    "    zmag = hdul[1].data['Mag_Z']\n",
    "    ymag = hdul[1].data['Mag_Y']\n",
    "    jmag = hdul[1].data['Mag_J']\n",
    "    hmag = hdul[1].data['Mag_H']\n",
    "    kmag = hdul[1].data['Mag_K']\n",
    "\n",
    "    temp_min = []\n",
    "    temp_mean = []\n",
    "    temp_max = []\n",
    "    mass_min = []\n",
    "    mass_mean = []\n",
    "    mass_max = []\n",
    "\n",
    "    for y in zmag:\n",
    "        temps = []\n",
    "\n",
    "        x = start_temp\n",
    "        while (zd_poly(x) + mu_near + zerr_poly(zd_poly(x) + mu_near)) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zd_poly(x) + mu_near + Az - zerr_poly(zd_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_near + zerr_poly(zc_poly(x) + mu_near) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_near + Az - zerr_poly(zc_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zd_poly(x) + mu_far + zerr_poly(zd_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zd_poly(x) + mu_far + Az - zerr_poly(zd_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_far + zerr_poly(zc_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_far + Az - zerr_poly(zc_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        \"\"\"x = start_temp\n",
    "        while zdo_poly(x) + mu_near + zerr_poly(zdo_poly(x) + mu_near) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_near + Az - zerr_poly(zdo_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_near + zerr_poly(zco_poly(x) + mu_near) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_near + Az - zerr_poly(zco_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_far + zerr_poly(zdo_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_far + Az - zerr_poly(zdo_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_far + zerr_poly(zco_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_far + Az - zerr_poly(zco_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\"\"\"\n",
    "        \n",
    "        temps_okay = []\n",
    "        \n",
    "        for alpha in temps:\n",
    "            if mass_poly(alpha) > 0.0:\n",
    "                temps_okay.append(alpha)\n",
    "                \n",
    "        temp_min.append(np.min(temps_okay))\n",
    "        temp_mean.append(np.mean(temps_okay))\n",
    "        temp_max.append(np.max(temps_okay))\n",
    "        \n",
    "        mass_okay = []\n",
    "        for beta in temps_okay:\n",
    "            mass_okay.append(mass_poly(beta))\n",
    "            \n",
    "        mass_min.append(np.min(mass_okay)) \n",
    "        mass_mean.append(np.mean(mass_okay))\n",
    "        mass_max.append(np.max(mass_okay))\n",
    "\n",
    "    col1 = fits.Column(name='RA_ICRS',format='D',unit='degree',array=RA_ICRS)\n",
    "    col2 = fits.Column(name='Dec_ICRS',format='D',unit='degree',array=Dec_ICRS)\n",
    "    col3 = fits.Column(name='Mag_Z',format='D',unit='mag',array=zmag)\n",
    "    col4 = fits.Column(name='Mag_Y',format='D',unit='mag',array=ymag)\n",
    "    col5 = fits.Column(name='Mag_J',format='D',unit='mag',array=jmag)\n",
    "    col6 = fits.Column(name='Mag_H',format='D',unit='mag',array=hmag)\n",
    "    col7 = fits.Column(name='Mag_K',format='D',unit='mag',array=kmag)\n",
    "    col8 = fits.Column(name='Min_Temp',format='F',unit='K',array=temp_min)\n",
    "    col9 = fits.Column(name='Mean_Temp',format='F',unit='K',array=temp_mean)\n",
    "    col10 = fits.Column(name='Max_Temp',format='F',unit='K',array=temp_max)\n",
    "    col11 = fits.Column(name='Min_Mass',format='F',unit='M_sol',array=mass_min)\n",
    "    col12 = fits.Column(name='Mean_Mass',format='F',unit='M_sol',array=mass_mean)\n",
    "    col14 = fits.Column(name='Max_Mass',format='F',unit='M_sol',array=mass_max)\n",
    "    \n",
    "    cols = fits.ColDefs([col1,col2,col3,col4,col5,col6,col7,col8,col9,col10,col11,col12,col14])\n",
    "    \n",
    "    '''Here we make the fits file'''\n",
    "\n",
    "    output_fits = output_name + '.fits'\n",
    "    data = fits.BinTableHDU.from_columns(cols)\n",
    "    data.writeto(output_fits)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def candidate_stats_old(file,output_name,start_temp=1200,max_temp=4500):\n",
    "    \n",
    "    '''This program finds the temperatures and masses of each object in the input file, and then creates\n",
    "    a table with that information. This version uses 5Myr set-ups.'''\n",
    "\n",
    "    hdul = fits.open(file)\n",
    "    \n",
    "    RA_ICRS = hdul[1].data['RA_ICRS']\n",
    "    Dec_ICRS = hdul[1].data['Dec_ICRS']\n",
    "    zmag = hdul[1].data['Mag_Z']\n",
    "    ymag = hdul[1].data['Mag_Y']\n",
    "    jmag = hdul[1].data['Mag_J']\n",
    "    hmag = hdul[1].data['Mag_H']\n",
    "    kmag = hdul[1].data['Mag_K']\n",
    "\n",
    "    temp_min = []\n",
    "    temp_mean = []\n",
    "    temp_max = []\n",
    "    mass_min = []\n",
    "    mass_mean = []\n",
    "    mass_max = []\n",
    "\n",
    "    for y in zmag:\n",
    "        temps = []\n",
    "\n",
    "        \"\"\"x = start_temp\n",
    "        while (zd_poly(x) + mu_near + zerr_poly(zd_poly(x) + mu_near)) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zd_poly(x) + mu_near + Az - zerr_poly(zd_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_near + zerr_poly(zc_poly(x) + mu_near) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_near + Az - zerr_poly(zc_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zd_poly(x) + mu_far + zerr_poly(zd_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zd_poly(x) + mu_far + Az - zerr_poly(zd_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_far + zerr_poly(zc_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_far + Az - zerr_poly(zc_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\"\"\"\n",
    "\n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_near + zerr_poly(zdo_poly(x) + mu_near) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_near + Az - zerr_poly(zdo_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_near + zerr_poly(zco_poly(x) + mu_near) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_near + Az - zerr_poly(zco_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_far + zerr_poly(zdo_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_far + Az - zerr_poly(zdo_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_far + zerr_poly(zco_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_far + Az - zerr_poly(zco_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "        \n",
    "        temps_okay = []\n",
    "        \n",
    "        for alpha in temps:\n",
    "            if masso_poly(alpha) > 0.0:\n",
    "                temps_okay.append(alpha)\n",
    "                \n",
    "        temp_min.append(np.min(temps_okay))\n",
    "        temp_mean.append(np.mean(temps_okay))\n",
    "        temp_max.append(np.max(temps_okay))\n",
    "        \n",
    "        mass_okay = []\n",
    "        for beta in temps_okay:\n",
    "            mass_okay.append(masso_poly(beta))\n",
    "            \n",
    "        mass_min.append(np.min(mass_okay)) \n",
    "        mass_mean.append(np.mean(mass_okay))\n",
    "        mass_max.append(np.max(mass_okay))\n",
    "\n",
    "    col1 = fits.Column(name='RA_ICRS',format='D',unit='degree',array=RA_ICRS)\n",
    "    col2 = fits.Column(name='Dec_ICRS',format='D',unit='degree',array=Dec_ICRS)\n",
    "    col3 = fits.Column(name='Mag_Z',format='D',unit='mag',array=zmag)\n",
    "    col4 = fits.Column(name='Mag_Y',format='D',unit='mag',array=ymag)\n",
    "    col5 = fits.Column(name='Mag_J',format='D',unit='mag',array=jmag)\n",
    "    col6 = fits.Column(name='Mag_H',format='D',unit='mag',array=hmag)\n",
    "    col7 = fits.Column(name='Mag_K',format='D',unit='mag',array=kmag)\n",
    "    col8 = fits.Column(name='Min_Temp',format='F',unit='K',array=temp_min)\n",
    "    col9 = fits.Column(name='Mean_Temp',format='F',unit='K',array=temp_mean)\n",
    "    col10 = fits.Column(name='Max_Temp',format='F',unit='K',array=temp_max)\n",
    "    col11 = fits.Column(name='Min_Mass',format='F',unit='M_sol',array=mass_min)\n",
    "    col12 = fits.Column(name='Mean_Mass',format='F',unit='M_sol',array=mass_mean)\n",
    "    col14 = fits.Column(name='Max_Mass',format='F',unit='M_sol',array=mass_max)\n",
    "    \n",
    "    cols = fits.ColDefs([col1,col2,col3,col4,col5,col6,col7,col8,col9,col10,col11,col12,col14])\n",
    "    \n",
    "    '''Here we make the fits file'''\n",
    "\n",
    "    output_fits = output_name + '.fits'\n",
    "    data = fits.BinTableHDU.from_columns(cols)\n",
    "    data.writeto(output_fits)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def candidate_stats_dusty(file,output_name,start_temp=1500,max_temp=4500):\n",
    "    \n",
    "    '''This program finds the temperatures and masses of each object in the input file, and then creates\n",
    "    a table with that information. This version uses Dusty set-ups.'''\n",
    "\n",
    "    hdul = fits.open(file)\n",
    "    \n",
    "    RA_ICRS = hdul[1].data['RA_ICRS']\n",
    "    Dec_ICRS = hdul[1].data['Dec_ICRS']\n",
    "    zmag = hdul[1].data['Mag_Z']\n",
    "    ymag = hdul[1].data['Mag_Y']\n",
    "    jmag = hdul[1].data['Mag_J']\n",
    "    hmag = hdul[1].data['Mag_H']\n",
    "    kmag = hdul[1].data['Mag_K']\n",
    "\n",
    "    temp_min = []\n",
    "    temp_mean = []\n",
    "    temp_max = []\n",
    "    mass_min = []\n",
    "    mass_mean_y = []\n",
    "    mass_mean_o = []\n",
    "    mass_max = []\n",
    "\n",
    "    for y in zmag:\n",
    "        temps = []\n",
    "\n",
    "        x = start_temp\n",
    "        while (zd_poly(x) + mu_near + zerr_poly(zd_poly(x) + mu_near)) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zd_poly(x) + mu_near + Az - zerr_poly(zd_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        \"\"\"x = start_temp\n",
    "        while zc_poly(x) + mu_near + zerr_poly(zc_poly(x) + mu_near) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_near + Az - zerr_poly(zc_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\"\"\"\n",
    "\n",
    "        x = start_temp\n",
    "        while zd_poly(x) + mu_far + zerr_poly(zd_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zd_poly(x) + mu_far + Az - zerr_poly(zd_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        \"\"\"x = start_temp\n",
    "        while zc_poly(x) + mu_far + zerr_poly(zc_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_far + Az - zerr_poly(zc_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\"\"\"\n",
    "\n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_near + zerr_poly(zdo_poly(x) + mu_near) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_near + Az - zerr_poly(zdo_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        \"\"\"x = start_temp\n",
    "        while zco_poly(x) + mu_near + zerr_poly(zco_poly(x) + mu_near) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_near + Az - zerr_poly(zco_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\"\"\"\n",
    "                \n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_far + zerr_poly(zdo_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_far + Az - zerr_poly(zdo_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        \"\"\"x = start_temp\n",
    "        while zco_poly(x) + mu_far + zerr_poly(zco_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_far + Az - zerr_poly(zco_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\"\"\"\n",
    "                \n",
    "        temp_min.append(np.min(temps))\n",
    "        temp_mean.append(np.mean(temps))\n",
    "        temp_max.append(np.max(temps))\n",
    "        mass_min.append(np.min([mass_poly(np.min(temps)),masso_poly(np.min(temps))]))\n",
    "        mass_mean_y.append(mass_poly(np.mean(temps)))\n",
    "        mass_mean_o.append(masso_poly(np.mean(temps)))\n",
    "        mass_max.append(np.max([mass_poly(np.max(temps)),masso_poly(np.max(temps))]))\n",
    "    \n",
    "    col1 = fits.Column(name='RA_ICRS',format='D',unit='degree',array=RA_ICRS)\n",
    "    col2 = fits.Column(name='Dec_ICRS',format='D',unit='degree',array=Dec_ICRS)\n",
    "    col3 = fits.Column(name='Mag_Z',format='D',unit='mag',array=zmag)\n",
    "    col4 = fits.Column(name='Mag_Y',format='D',unit='mag',array=ymag)\n",
    "    col5 = fits.Column(name='Mag_J',format='D',unit='mag',array=jmag)\n",
    "    col6 = fits.Column(name='Mag_H',format='D',unit='mag',array=hmag)\n",
    "    col7 = fits.Column(name='Mag_K',format='D',unit='mag',array=kmag)\n",
    "    col8 = fits.Column(name='Min_Temp',format='F',unit='K',array=temp_min)\n",
    "    col9 = fits.Column(name='Mean_Temp',format='F',unit='K',array=temp_mean)\n",
    "    col10 = fits.Column(name='Max_Temp',format='F',unit='K',array=temp_max)\n",
    "    col11 = fits.Column(name='Min_Mass',format='F',unit='M_sol',array=mass_min)\n",
    "    col12 = fits.Column(name='Mean_Mass_Young',format='F',unit='M_sol',array=mass_mean_y)\n",
    "    col13 = fits.Column(name='Mean_Mass_Old',format='F',unit='M_sol',array=mass_mean_o)\n",
    "    col14 = fits.Column(name='Max_Mass',format='F',unit='M_sol',array=mass_max)\n",
    "    \n",
    "    cols = fits.ColDefs([col1,col2,col3,col4,col5,col6,col7,col8,col9,col10,col11,col12,col13,col14])\n",
    "    \n",
    "    '''Here we make the fits file'''\n",
    "\n",
    "    output_fits = output_name + '.fits'\n",
    "    data = fits.BinTableHDU.from_columns(cols)\n",
    "    data.writeto(output_fits)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def candidate_stats_cond(file,output_name,start_temp=900,max_temp=4500):\n",
    "    \n",
    "    '''This program finds the temperatures and masses of each object in the input file, and then creates\n",
    "    a table with that information. This version uses Cond set-ups.'''\n",
    "\n",
    "    hdul = fits.open(file)\n",
    "    \n",
    "    RA_ICRS = hdul[1].data['RA_ICRS']\n",
    "    Dec_ICRS = hdul[1].data['Dec_ICRS']\n",
    "    zmag = hdul[1].data['Mag_Z']\n",
    "    ymag = hdul[1].data['Mag_Y']\n",
    "    jmag = hdul[1].data['Mag_J']\n",
    "    hmag = hdul[1].data['Mag_H']\n",
    "    kmag = hdul[1].data['Mag_K']\n",
    "\n",
    "    temp_min = []\n",
    "    temp_mean = []\n",
    "    temp_max = []\n",
    "    mass_min = []\n",
    "    mass_mean_y = []\n",
    "    mass_mean_o = []\n",
    "    mass_max = []\n",
    "\n",
    "    for y in zmag:\n",
    "        temps = []\n",
    "\n",
    "        \"\"\"x = start_temp\n",
    "        while (zd_poly(x) + mu_near + zerr_poly(zd_poly(x) + mu_near)) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zd_poly(x) + mu_near + Az - zerr_poly(zd_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\"\"\"\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_near + zerr_poly(zc_poly(x) + mu_near) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_near + Az - zerr_poly(zc_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        \"\"\"x = start_temp\n",
    "        while zd_poly(x) + mu_far + zerr_poly(zd_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zd_poly(x) + mu_far + Az - zerr_poly(zd_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\"\"\"\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_far + zerr_poly(zc_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zc_poly(x) + mu_far + Az - zerr_poly(zc_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        \"\"\"x = start_temp\n",
    "        while zdo_poly(x) + mu_near + zerr_poly(zdo_poly(x) + mu_near) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "\n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_near + Az - zerr_poly(zdo_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\"\"\"\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_near + zerr_poly(zco_poly(x) + mu_near) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_near + Az - zerr_poly(zco_poly(x) + mu_near + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        \"\"\"x = start_temp\n",
    "        while zdo_poly(x) + mu_far + zerr_poly(zdo_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zdo_poly(x) + mu_far + Az - zerr_poly(zdo_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\"\"\"\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_far + zerr_poly(zco_poly(x) + mu_far) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "                \n",
    "        x = start_temp\n",
    "        while zco_poly(x) + mu_far + Az - zerr_poly(zco_poly(x) + mu_far + Az) > y and x < max_temp:\n",
    "            x = x+1    \n",
    "        if x != start_temp:\n",
    "            if x != max_temp:\n",
    "                temps.append(x)\n",
    "         \n",
    "        if temps != []:\n",
    "            temp_min.append(np.min(temps))\n",
    "            temp_mean.append(np.mean(temps))\n",
    "            temp_max.append(np.max(temps))\n",
    "            mass_min.append(np.min([mass_poly(np.min(temps)),masso_poly(np.min(temps))]))\n",
    "            mass_mean_y.append(mass_poly(np.mean(temps)))\n",
    "            mass_mean_o.append(masso_poly(np.mean(temps)))\n",
    "            mass_max.append(np.max([mass_poly(np.max(temps)),masso_poly(np.max(temps))]))\n",
    "    \n",
    "    col1 = fits.Column(name='RA_ICRS',format='D',unit='degree',array=RA_ICRS)\n",
    "    col2 = fits.Column(name='Dec_ICRS',format='D',unit='degree',array=Dec_ICRS)\n",
    "    col3 = fits.Column(name='Mag_Z',format='D',unit='mag',array=zmag)\n",
    "    col4 = fits.Column(name='Mag_Y',format='D',unit='mag',array=ymag)\n",
    "    col5 = fits.Column(name='Mag_J',format='D',unit='mag',array=jmag)\n",
    "    col6 = fits.Column(name='Mag_H',format='D',unit='mag',array=hmag)\n",
    "    col7 = fits.Column(name='Mag_K',format='D',unit='mag',array=kmag)\n",
    "    col8 = fits.Column(name='Min_Temp',format='F',unit='K',array=temp_min)\n",
    "    col9 = fits.Column(name='Mean_Temp',format='F',unit='K',array=temp_mean)\n",
    "    col10 = fits.Column(name='Max_Temp',format='F',unit='K',array=temp_max)\n",
    "    col11 = fits.Column(name='Min_Mass',format='F',unit='M_sol',array=mass_min)\n",
    "    col12 = fits.Column(name='Mean_Mass_Young',format='F',unit='M_sol',array=mass_mean_y)\n",
    "    col13 = fits.Column(name='Mean_Mass_Old',format='F',unit='M_sol',array=mass_mean_o)\n",
    "    col14 = fits.Column(name='Max_Mass',format='F',unit='M_sol',array=mass_max)\n",
    "    \n",
    "    cols = fits.ColDefs([col1,col2,col3,col4,col5,col6,col7,col8,col9,col10,col11,col12,col13,col14])\n",
    "    \n",
    "    '''Here we make the fits file'''\n",
    "\n",
    "    output_fits = output_name + '.fits'\n",
    "    data = fits.BinTableHDU.from_columns(cols)\n",
    "    data.writeto(output_fits)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "candidate_stats_dusty('Final_Progress/final_working_0821.fits','temp_mass_dusty')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "candidate_stats_cond('Final_Progress/final_working_0821.fits','temp_mass_cond')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "candidate_stats('Final_Progress/final_working_0821.fits','0904_temp_mass')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "candidate_stats_old('Final_Progress/final_working_0821.fits','0909_temp_mass_old')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "scrolled": false
   },
   "outputs": [],
   "source": [
    "candidate_stats_young('Final_Progress/final_working_0821.fits','0909_temp_mass_young')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "'''The next few cells were used to calculate the area and density of the field of view and a couple\n",
    "other important areas. It is just a lot of simple math.'''\n",
    "galaxy_x_1 = 168.842\n",
    "galaxy_x_2 = 168.843\n",
    "galaxy_x_3 = 167.373\n",
    "galaxy_x_4 = 167.374\n",
    "galaxy_y_1 = -2.051\n",
    "galaxy_y_2 = -0.851\n",
    "galaxy_y_3 = -0.856\n",
    "galaxy_y_4 = -2.053\n",
    "field_x_1 = 85.1201\n",
    "field_x_2 = 82.8387\n",
    "field_x_3 = 82.273\n",
    "field_x_4 = 84.5136\n",
    "field_y_1 = 0.0260\n",
    "field_y_2 = 0.6387\n",
    "field_y_3 = -1.591\n",
    "field_y_4 = -2.1839\n",
    "cut1_x_1 = 83.4215\n",
    "cut1_x_2 = 83.5151\n",
    "cut1_x_3 = 83.5831\n",
    "cut1_x_4 = 83.487\n",
    "cut1_y_1 = -1.8838\n",
    "cut1_y_2 = -1.5260\n",
    "cut1_y_3 = -1.5413\n",
    "cut1_y_4 = -1.902\n",
    "cut2_x_1 = 83.585\n",
    "cut2_x_2 = 83.653\n",
    "cut2_x_3 = 83.7503\n",
    "cut2_x_4 = 83.6846\n",
    "cut2_y_1 = -1.930\n",
    "cut2_y_2 = -1.947\n",
    "cut2_y_3 = -1.5885\n",
    "cut2_y_4 = -1.5720\n",
    "cut3_x_1 = 83.7976\n",
    "cut3_x_2 = 83.7333\n",
    "cut3_x_3 = 83.7961\n",
    "cut3_x_4 = 83.8634\n",
    "cut3_y_1 = -0.7389\n",
    "cut3_y_2 = -0.7105\n",
    "cut3_y_3 = -0.4851\n",
    "cut3_y_4 = -0.4995\n",
    "cut4_x_1 = 83.9645\n",
    "cut4_x_2 = 84.0329\n",
    "cut4_x_3 = 83.9677\n",
    "cut4_x_4 = 83.8942\n",
    "cut4_y_1 = -0.5265\n",
    "cut4_y_2 = -0.5457\n",
    "cut4_y_3 = -0.7834\n",
    "cut4_y_4 = -0.7623\n",
    "field2_x_1 = 82.564\n",
    "field2_x_2 = field_x_3\n",
    "field2_x_3 = 83.672\n",
    "field2_x_4 = 83.988\n",
    "field2_y_1 = -0.407\n",
    "field2_y_2 = field_y_3\n",
    "field2_y_3 = -1.953\n",
    "field2_y_4 = -0.789\n",
    "field3_x_1 = 83.656\n",
    "field3_x_2 = 83.969\n",
    "field3_x_3 = 84.839\n",
    "field3_x_4 = field_x_4\n",
    "field3_y_1 = -1.948\n",
    "field3_y_2 = -0.783\n",
    "field3_y_3 = -1.0164\n",
    "field3_y_4 = field_y_4\n",
    "field6_x_1 = 82.528\n",
    "field6_x_2 = field_x_2\n",
    "field6_x_3 = 84.267\n",
    "field6_x_4 = 83.953\n",
    "field6_y_1 = -0.524\n",
    "field6_y_2 = field_y_2\n",
    "field6_y_3 = 0.255\n",
    "field6_y_4 = -0.911\n",
    "field7_x_1 = 83.936\n",
    "field7_x_2 = 84.248\n",
    "field7_x_3 = field_x_1\n",
    "field7_x_4 = 84.806\n",
    "field7_y_1 = -0.902\n",
    "field7_y_2 = 0.259\n",
    "field7_y_3 = field_y_1\n",
    "field7_y_4 = -1.140"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "galaxy_area = abs(((galaxy_x_1*galaxy_y_2 - galaxy_y_1 * galaxy_x_2) + \n",
    " (galaxy_x_2*galaxy_y_3 - galaxy_y_2 * galaxy_x_3) + \n",
    " (galaxy_x_3*galaxy_y_4 - galaxy_y_3 * galaxy_x_4) + \n",
    " (galaxy_x_4*galaxy_y_1 - galaxy_y_4 * galaxy_x_1))/2)\n",
    "print(galaxy_area)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "field_area = abs(((field_x_1*field_y_2 - field_y_1 * field_x_2) + \n",
    " (field_x_2*field_y_3 - field_y_2 * field_x_3) + \n",
    " (field_x_3*field_y_4 - field_y_3 * field_x_4) + \n",
    " (field_x_4*field_y_1 - field_y_4 * field_x_1))/2)\n",
    "print(field_area)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "cut1_area = abs(((cut1_x_1*cut1_y_2 - cut1_y_1 * cut1_x_2) + \n",
    " (cut1_x_2*cut1_y_3 - cut1_y_2 * cut1_x_3) + \n",
    " (cut1_x_3*cut1_y_4 - cut1_y_3 * cut1_x_4) + \n",
    " (cut1_x_4*cut1_y_1 - cut1_y_4 * cut1_x_1))/2)\n",
    "print(cut1_area)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "cut2_area = abs(((cut2_x_1*cut2_y_2 - cut2_y_1 * cut2_x_2) + \n",
    " (cut2_x_2*cut2_y_3 - cut2_y_2 * cut2_x_3) + \n",
    " (cut2_x_3*cut2_y_4 - cut2_y_3 * cut2_x_4) + \n",
    " (cut2_x_4*cut2_y_1 - cut2_y_4 * cut2_x_1))/2)\n",
    "print(cut2_area)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "cut3_area = abs(((cut3_x_1*cut3_y_2 - cut3_y_1 * cut3_x_2) + \n",
    " (cut3_x_2*cut3_y_3 - cut3_y_2 * cut3_x_3) + \n",
    " (cut3_x_3*cut3_y_4 - cut3_y_3 * cut3_x_4) + \n",
    " (cut3_x_4*cut3_y_1 - cut3_y_4 * cut3_x_1))/2)\n",
    "print(cut3_area)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "cut4_area = abs(((cut4_x_1*cut4_y_2 - cut4_y_1 * cut4_x_2) + \n",
    " (cut4_x_2*cut4_y_3 - cut4_y_2 * cut4_x_3) + \n",
    " (cut4_x_3*cut4_y_4 - cut4_y_3 * cut4_x_4) + \n",
    " (cut4_x_4*cut4_y_1 - cut4_y_4 * cut4_x_1))/2)\n",
    "print(cut4_area)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "field2_area = abs(((field2_x_1*field2_y_2 - field2_y_1 * field2_x_2) + \n",
    " (field2_x_2*field2_y_3 - field2_y_2 * field2_x_3) + \n",
    " (field2_x_3*field2_y_4 - field2_y_3 * field2_x_4) + \n",
    " (field2_x_4*field2_y_1 - field2_y_4 * field2_x_1))/2) - cut1_area - cut2_area\n",
    "print(field2_area)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "field3_area = abs(((field3_x_1*field3_y_2 - field3_y_1 * field3_x_2) + \n",
    " (field3_x_2*field3_y_3 - field3_y_2 * field3_x_3) + \n",
    " (field3_x_3*field3_y_4 - field3_y_3 * field3_x_4) + \n",
    " (field3_x_4*field3_y_1 - field3_y_4 * field3_x_1))/2)\n",
    "print(field3_area)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "field6_area = abs(((field6_x_1*field6_y_2 - field6_y_1 * field6_x_2) + \n",
    " (field6_x_2*field6_y_3 - field6_y_2 * field6_x_3) + \n",
    " (field6_x_3*field6_y_4 - field6_y_3 * field6_x_4) + \n",
    " (field6_x_4*field6_y_1 - field6_y_4 * field6_x_1))/2) - cut3_area - cut4_area\n",
    "print(field6_area)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "field7_area = abs(((field7_x_1*field7_y_2 - field7_y_1 * field7_x_2) + \n",
    " (field7_x_2*field7_y_3 - field7_y_2 * field7_x_3) + \n",
    " (field7_x_3*field7_y_4 - field7_y_3 * field7_x_4) + \n",
    " (field7_x_4*field7_y_1 - field7_y_4 * field7_x_1))/2)\n",
    "print(field7_area)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "final_field_area = field_area - cut1_area - cut2_area - cut3_area - cut4_area "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "print(galaxy_area)\n",
    "print(final_field_area)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "'''I start calculating densities with the obtained areas.'''\n",
    "\n",
    "ipmo = 114\n",
    "pos_ipmo = 71\n",
    "bd = 89\n",
    "pos_bd = 520\n",
    "yso = 154\n",
    "likely_ipmo = 160\n",
    "likely_bd = 274\n",
    "likely_yso = 514"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "likely_yso/final_field_area"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "final_field_area_sr = final_field_area/3283\n",
    "print(final_field_area_sr)\n",
    "r2 = 75\n",
    "r1 = 25\n",
    "V1 = (final_field_area_sr/3)*(r1**3)\n",
    "print(V1)\n",
    "V2 = (final_field_area_sr/3)*(r2**3)\n",
    "print(V2)\n",
    "V_T = (final_field_area_sr/3)*(r2**3 - r1**3)\n",
    "print(V_T)\n",
    "cont_min = 2*10**(-3)\n",
    "cont_max = 8*10**(-3)\n",
    "print(V_T*cont_min)\n",
    "print(V_T*cont_max)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "galaxy_density = 3/galaxy_area"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "galaxy_density"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "galaxy_density * final_field_area"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "galaxy_density"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "caballero_size = 1.5*1.5\n",
    "print(caballero_size)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "cab_final_size = caballero_size*1.75"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "cab_galaxy_density = 11/cab_final_size"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "cab_galaxy_density * final_field_area"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "final_field_area"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "total = 948\n",
    "ipmo = 112\n",
    "poss_ipmo = 62\n",
    "bd = 98\n",
    "poss_bd = 490\n",
    "yso = 186\n",
    "total_2 = 383\n",
    "ipmo_2 = 27\n",
    "poss_ipmo_2 = 30\n",
    "bd_2 = 39\n",
    "poss_bd_2 = 209\n",
    "yso_2 = 78\n",
    "total_3 = 255\n",
    "ipmo_3 = 23\n",
    "poss_ipmo_3 = 13\n",
    "bd_3 = 27\n",
    "poss_bd_3 = 139\n",
    "yso_3 = 53\n",
    "total_6 = 260\n",
    "ipmo_6 = 42\n",
    "poss_ipmo_6 = 16\n",
    "bd_6 = 38\n",
    "poss_bd_6 = 118\n",
    "yso_6 = 46\n",
    "total_7 = 181\n",
    "ipmo_7 = 34\n",
    "poss_ipmo_7 = 11\n",
    "bd_7 = 19\n",
    "poss_bd_7 = 83\n",
    "yso_7 = 34"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "total_total_density = total/final_field_area\n",
    "ipmo_total_density = (ipmo-2)/final_field_area\n",
    "possipmo_total_density = poss_ipmo/final_field_area\n",
    "bd_total_density = bd/final_field_area\n",
    "possbd_total_density = poss_bd/final_field_area\n",
    "yso_total_density = yso/final_field_area\n",
    "comb_ipmo_total_density = (ipmo + poss_ipmo)/final_field_area\n",
    "\n",
    "total_2_density = total_2/field2_area\n",
    "ipmo_2_density = ipmo_2/field2_area\n",
    "possipmo_2_density = poss_ipmo_2/field2_area\n",
    "bd_2_density = bd_2/field2_area\n",
    "possbd_2_density = poss_bd_2/field2_area\n",
    "yso_2_density = yso_2/field2_area\n",
    "comb_ipmo_2_density = (ipmo_2 + poss_ipmo_2)/field2_area\n",
    "\n",
    "total_3_density = total_3/field3_area\n",
    "ipmo_3_density = ipmo_3/field3_area\n",
    "possipmo_3_density = poss_ipmo_3/field3_area\n",
    "bd_3_density = bd_3/field3_area\n",
    "possbd_3_density = poss_bd_3/field3_area\n",
    "yso_3_density = yso_3/field3_area\n",
    "comb_ipmo_3_density = (ipmo_3 + poss_ipmo_3)/field3_area\n",
    "\n",
    "total_6_density = total_6/field6_area\n",
    "ipmo_6_density = ipmo_6/field6_area\n",
    "possipmo_6_density = poss_ipmo_6/field6_area\n",
    "bd_6_density = bd_6/field6_area\n",
    "possbd_6_density = poss_bd_6/field6_area\n",
    "yso_6_density = yso_6/field6_area\n",
    "comb_ipmo_6_density = (ipmo_6 + poss_ipmo_6)/field6_area\n",
    "\n",
    "total_7_density = total_7/field7_area\n",
    "ipmo_7_density = ipmo_7/field7_area\n",
    "possipmo_7_density = poss_ipmo_7/field7_area\n",
    "bd_7_density = bd_7/field7_area\n",
    "possbd_7_density = poss_bd_7/field7_area\n",
    "yso_7_density = yso_7/field7_area\n",
    "comb_ipmo_7_density = (ipmo_7 + poss_ipmo_7)/field7_area\n",
    "\n",
    "peak_density = ipmo_total_density\n",
    "ipmo_density = comb_ipmo_total_density\n",
    "bd_density = possipmo_total_density + bd_total_density + possbd_total_density\n",
    "yso_density = possbd_total_density + yso_total_density"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "print(total_total_density)\n",
    "print(peak_density)\n",
    "print(ipmo_density)\n",
    "print(bd_density)\n",
    "print(yso_density)\n",
    "print(ipmo-2)\n",
    "print(ipmo + poss_ipmo)\n",
    "print(poss_ipmo + bd + poss_bd)\n",
    "print(poss_bd + yso)\n",
    "print(total)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "print(total_total_density)\n",
    "print(total_2_density)\n",
    "print(total_3_density)\n",
    "print(total_6_density)\n",
    "print(total_7_density)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "print(ipmo_total_density)\n",
    "print(ipmo_2_density)\n",
    "print(ipmo_3_density)\n",
    "print(ipmo_6_density)\n",
    "print(ipmo_7_density)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "print(possipmo_total_density)\n",
    "print(possipmo_2_density)\n",
    "print(possipmo_3_density)\n",
    "print(possipmo_6_density)\n",
    "print(possipmo_7_density)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "print(bd_total_density)\n",
    "print(bd_2_density)\n",
    "print(bd_3_density)\n",
    "print(bd_6_density)\n",
    "print(bd_7_density)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "print(possbd_total_density)\n",
    "print(possbd_2_density)\n",
    "print(possbd_3_density)\n",
    "print(possbd_6_density)\n",
    "print(possbd_7_density)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "print(yso_total_density)\n",
    "print(yso_2_density)\n",
    "print(yso_3_density)\n",
    "print(yso_6_density)\n",
    "print(yso_7_density)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "print(ipmo_total_density + possipmo_total_density)\n",
    "print(ipmo_2_density + possipmo_2_density)\n",
    "print(ipmo_3_density + possipmo_3_density)\n",
    "print(ipmo_6_density + possipmo_6_density)\n",
    "print(ipmo_7_density + possipmo_7_density)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def vosa(file,output_name):\n",
    "    \n",
    "    '''This program takes a fits file and makes it into a form that can then be saved as a cvs file,\n",
    "    which is then ready to be used on the VOSA website, to make it into their format.'''\n",
    "\n",
    "    hdul = fits.open(file)\n",
    "    \n",
    "    Ob_ID = hdul[1].data['Object_ID']\n",
    "    RA_ICRS = hdul[1].data['RA_ICRS']\n",
    "    Dec_ICRS = hdul[1].data['Dec_ICRS']\n",
    "    zmag = hdul[1].data['Mag_Z']\n",
    "    ymag = hdul[1].data['Mag_Y']\n",
    "    jmag = hdul[1].data['Mag_J']\n",
    "    hmag = hdul[1].data['Mag_H']\n",
    "    kmag = hdul[1].data['Mag_K']\n",
    "    av = hdul[1].data['Av']\n",
    "    distance = hdul[1].data['Distance']\n",
    "    \n",
    "    zmag_err = zerr_poly(zmag)\n",
    "    ymag_err = yerr_poly(ymag)\n",
    "    jmag_err = jerr_poly(jmag)\n",
    "    hmag_err = herr_poly(hmag)\n",
    "    kmag_err = kerr_poly(kmag)\n",
    "    \n",
    "    col1 = fits.Column(name='RA_ICRS',format='D',unit='degree',array=RA_ICRS)\n",
    "    col2 = fits.Column(name='Dec_ICRS',format='D',unit='degree',array=Dec_ICRS)\n",
    "    col3 = fits.Column(name='Mag_Z',format='D',unit='mag',array=zmag)\n",
    "    col3a = fits.Column(name='Mag_Z_Error',format='D',unit='mag',array=zmag_err)\n",
    "    col4 = fits.Column(name='Mag_Y',format='D',unit='mag',array=ymag)\n",
    "    col4a = fits.Column(name='Mag_Y_Error',format='D',unit='mag',array=ymag_err)\n",
    "    col5 = fits.Column(name='Mag_J',format='D',unit='mag',array=jmag)\n",
    "    col5a = fits.Column(name='Mag_J_Error',format='D',unit='mag',array=jmag_err)\n",
    "    col6 = fits.Column(name='Mag_H',format='D',unit='mag',array=hmag)\n",
    "    col6a = fits.Column(name='Mag_H_Error',format='D',unit='mag',array=hmag_err)\n",
    "    col7 = fits.Column(name='Mag_K',format='D',unit='mag',array=kmag)\n",
    "    col7a = fits.Column(name='Mag_K_Error',format='D',unit='mag',array=kmag_err)\n",
    "    col8 = fits.Column(name='Av',format='10A',array=av)\n",
    "    col9 = fits.Column(name='Distance',format='I',array=distance)\n",
    "    col10 = fits.Column(name='Object_ID',format='I',array=Ob_ID)\n",
    "    \n",
    "    cols = fits.ColDefs([col10,col1,col2,col3,col3a,col4,col4a,col5,col5a,col6,col6a,col7,col7a,col8,col9])\n",
    "    \n",
    "    '''Here we make the fits file'''\n",
    "\n",
    "    output_fits = output_name + '.fits'\n",
    "    data = fits.BinTableHDU.from_columns(cols)\n",
    "    data.writeto(output_fits)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "vosa('vosa_ready_full.fits','vosa_ready_full_01')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "m_sol = 1.98847*10**30\n",
    "m_jup = 1.89813*10**27"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "def make_table(file,output_name):\n",
    "    \n",
    "    '''This takes a fits file and turns it into one that can be used to easily make Latex tables.'''\n",
    "\n",
    "    hdul = fits.open(file)\n",
    "    \n",
    "    Ob_ID = hdul[1].data['Object_ID']\n",
    "    RA_ICRS = hdul[1].data['RA_dms']\n",
    "    Dec_ICRS = hdul[1].data['Dec_dms']\n",
    "    zmag = hdul[1].data['Mag_Z']\n",
    "    ymag = hdul[1].data['Mag_Y']\n",
    "    jmag = hdul[1].data['Mag_J']\n",
    "    hmag = hdul[1].data['Mag_H']\n",
    "    kmag = hdul[1].data['Mag_K']\n",
    "    filters = hdul[1].data['Filters']\n",
    "    counts = hdul[1].data['Total Counts']\n",
    "    perc = hdul[1].data['Percentage Count']\n",
    "    temp_min = hdul[1].data['Min_Temp']\n",
    "    temp_mean = hdul[1].data['Mean_Temp']\n",
    "    temp_max = hdul[1].data['Max_Temp']\n",
    "    mass_min = hdul[1].data['Min_Mass']\n",
    "    mass_mean = hdul[1].data['Mean_Mass']\n",
    "    mass_max = hdul[1].data['Max_Mass']\n",
    "    \n",
    "    allpoints = list(range(len(hdul[1].data)))\n",
    "    \n",
    "    Dec_fixed = [0 for z in allpoints]\n",
    "    \n",
    "    zmag = np.around(zmag,decimals=2)\n",
    "    ymag = np.around(ymag,decimals=2)\n",
    "    jmag = np.around(jmag,decimals=2)\n",
    "    hmag = np.around(hmag,decimals=2)\n",
    "    kmag = np.around(kmag,decimals=2)\n",
    "    perc = np.around(perc,decimals=3)\n",
    "    mass_min = np.around(mass_min,decimals=5)\n",
    "    mass_mean = np.around(mass_mean,decimals=5)\n",
    "    mass_max = np.around(mass_max,decimals=5)\n",
    "    \n",
    "    for x in range(len(allpoints)):\n",
    "        Dec_fixed[x] = Dec_ICRS[x][0]+Dec_ICRS[x][2]+'$^{\\circ}$ '+Dec_ICRS[x][4:6]+'\\' '+Dec_ICRS[x][7:11]+'\\\" '\n",
    "    \n",
    "    \n",
    "    col0 = fits.Column(name='Object_ID',format='I',array=Ob_ID)\n",
    "    col1 = fits.Column(name='RA_ICRS',format='11A',unit='degree',array=RA_ICRS)\n",
    "    col2 = fits.Column(name='Dec_ICRS',format='132A',unit='degree',array=Dec_fixed)\n",
    "    col3 = fits.Column(name='Mag_Z',format='D2',unit='mag',array=zmag)\n",
    "    col4 = fits.Column(name='Mag_Y',format='D2',unit='mag',array=ymag)\n",
    "    col5 = fits.Column(name='Mag_J',format='D2',unit='mag',array=jmag)\n",
    "    col6 = fits.Column(name='Mag_H',format='D2',unit='mag',array=hmag)\n",
    "    col7 = fits.Column(name='Mag_K',format='D2',unit='mag',array=kmag)\n",
    "    col8 = fits.Column(name='Filters',format='I',unit='Counts',array=filters)\n",
    "    col9 = fits.Column(name='Counts',format='I',unit='Counts',array=counts)\n",
    "    col10 = fits.Column(name='Perc',format='D2',unit='Percentage',array=perc)\n",
    "    col11 = fits.Column(name='Min_Temp',format='I',unit='K',array=temp_min)\n",
    "    col12 = fits.Column(name='Mean_Temp',format='I',unit='K',array=temp_mean)\n",
    "    col13 = fits.Column(name='Max_Temp',format='I',unit='K',array=temp_max)\n",
    "    col14 = fits.Column(name='Min_Mass',format='E5',unit='M_sol',array=mass_min)\n",
    "    col15 = fits.Column(name='Mean_Mass',format='E5',unit='M_sol',array=mass_mean)\n",
    "    col17 = fits.Column(name='Max_Mass',format='E5',unit='M_sol',array=mass_max)\n",
    "    \n",
    "    cols = fits.ColDefs([col0,col1,col2,col3,col4,col5,col6,col7,col8,col9,\n",
    "                         col10,col11,col12,col13,col14,col15,col17])\n",
    "    \n",
    "    '''Here we make the fits file'''\n",
    "\n",
    "    output_fits = output_name + '.fits'\n",
    "    data = fits.BinTableHDU.from_columns(cols)\n",
    "    data.writeto(output_fits)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "make_table('Final_Progress/final_old.fits','final_table_old')\n",
    "make_table('Final_Progress/final_young.fits','final_table_young')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "'''The next few cells are used to make the density plots.'''\n",
    "hdul = fits.open('Final_Progress/final.fits')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "x_cand = hdul[1].data['RA_ICRS']\n",
    "y_cand = hdul[1].data['Dec_ICRS']\n",
    "x_small = hdul[1].data['RA_ICRS'][0:104]\n",
    "y_small = hdul[1].data['Dec_ICRS'][0:104]\n",
    "x_ipmo = hdul[1].data['RA_ICRS'][0:175]\n",
    "y_ipmo = hdul[1].data['Dec_ICRS'][0:175]\n",
    "x_bd = hdul[1].data['RA_ICRS'][115:763]\n",
    "y_bd = hdul[1].data['Dec_ICRS'][115:763]\n",
    "x_yso = hdul[1].data['RA_ICRS'][274:-1]\n",
    "y_yso = hdul[1].data['Dec_ICRS'][274:-1]\n",
    "\n",
    "x_small_or = x_small\n",
    "y_small_or = y_small"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "densities_x = [83.20233047,83.95468407,83.50452033,84.12083681]\n",
    "densities_y = [-1.64303844,-1.56590665,-0.72506968,-0.39312178]\n",
    "n = ['D','A', 'C','B']\n",
    "\n",
    "r = .67/2\n",
    "circ_area=np.pi * r**2"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "xy = np.vstack([x_cand,y_cand])\n",
    "z = gaussian_kde(xy)(xy)\n",
    "\n",
    "idx = z.argsort()\n",
    "x, y, z = x_cand[idx], y_cand[idx], z[idx]\n",
    "\n",
    "fig, ax = plt.subplots()\n",
    "fig.set_size_inches(15,9)\n",
    "ax.set_facecolor((64/255, 3/255, 80/255))\n",
    "im = ax.scatter(x, y, c=z, s=10000, edgecolor='')\n",
    "v = np.linspace(-.1, 2.0, 100, endpoint=True)\n",
    "cb = plt.colorbar(im,fraction=0.05)\n",
    "cb.ax.set_yticklabels([int(0.05*(total_total_density/circ_area)),int(0.10*(total_total_density/circ_area)),\n",
    "                       int(0.15*(total_total_density/circ_area)),int(0.20*(total_total_density/circ_area)),\n",
    "                       int(0.25*(total_total_density/circ_area)),int(0.30*(total_total_density/circ_area))],size=16)\n",
    "cb.set_label('Surface Density per Square Degree',size=20)\n",
    "#ax.scatter(x, y, color='k',s=1)\n",
    "ax.scatter(84.05224370308734,-1.1913836528004638, color='r', marker ='*', s=120)\n",
    "ax.scatter(densities_x,densities_y, color='b', marker = 'D', s=24)\n",
    "#for i, txt in enumerate(n):\n",
    "#    ax.annotate(txt, (densities_x[i], densities_y[i]))\n",
    "plt.title('Candidate Density',fontsize=36)\n",
    "plt.xlabel('RA', fontsize=28)\n",
    "plt.ylabel('Dec', fontsize=28)\n",
    "plt.gca().invert_xaxis()\n",
    "plt.savefig('cand_density_plot_04')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "xy = np.vstack([x_small,y_small])\n",
    "z = gaussian_kde(xy)(xy)\n",
    "\n",
    "idx = z.argsort()\n",
    "x, y, z = x_small[idx], y_small[idx], z[idx]\n",
    "\n",
    "fig, ax = plt.subplots()\n",
    "fig.set_size_inches(15,9)\n",
    "ax.set_facecolor((64/255, 3/255, 80/255))\n",
    "im = ax.scatter(x, y, c=z, s=10000, edgecolor='')\n",
    "cb = plt.colorbar(im,fraction=0.046)\n",
    "cb.set_label('Surface Density per Square Degree',size=20)\n",
    "cb.ax.set_yticklabels([int(0.05*(ipmo_total_density/circ_area)),int(0.10*(ipmo_total_density/circ_area)),\n",
    "                       int(0.15*(ipmo_total_density/circ_area)),int(0.20*(ipmo_total_density/circ_area)),\n",
    "                       int(0.25*(ipmo_total_density/circ_area)),int(0.30*(ipmo_total_density/circ_area))],size=16)\n",
    "#ax.scatter(x_small_or, y_small_or, color='k',s=1)\n",
    "ax.scatter(84.05224370308734,-1.1913836528004638, color='r', marker ='*', s=120)\n",
    "plt.title('Peak Candidate Density',fontsize=36)\n",
    "plt.xlabel('RA', fontsize=28)\n",
    "plt.ylabel('Dec', fontsize=28)\n",
    "plt.gca().invert_xaxis()\n",
    "plt.savefig('smallest_density_plot_04')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "xy = np.vstack([x_ipmo,y_ipmo])\n",
    "z = gaussian_kde(xy)(xy)\n",
    "\n",
    "idx = z.argsort()\n",
    "x, y, z = x_ipmo[idx], y_ipmo[idx], z[idx]\n",
    "\n",
    "fig, ax = plt.subplots()\n",
    "fig.set_size_inches(15,9)\n",
    "ax.set_facecolor((64/255, 3/255, 80/255))\n",
    "im = ax.scatter(x, y, c=z, s=10000, edgecolor='')\n",
    "cb = plt.colorbar(im,fraction=0.046)\n",
    "cb.set_label('Surface Density per Square Degree',size=20)\n",
    "cb.ax.set_yticklabels([int(0.05*(comb_ipmo_total_density/circ_area)),int(0.10*(comb_ipmo_total_density/circ_area)),\n",
    "                       int(0.15*(comb_ipmo_total_density/circ_area)),int(0.20*(comb_ipmo_total_density/circ_area)),\n",
    "                       int(0.25*(comb_ipmo_total_density/circ_area)),int(0.30*(comb_ipmo_total_density/circ_area))],size=16)\n",
    "#ax.scatter(x_ipmo_or, y_ipmo_or, color='k',s=1)\n",
    "ax.scatter(84.05224370308734,-1.1913836528004638, color='r', marker ='*', s=120)\n",
    "plt.title('Candidate iPMO Density',fontsize=36)\n",
    "plt.xlabel('RA', fontsize=28)\n",
    "plt.ylabel('Dec', fontsize=28)\n",
    "plt.gca().invert_xaxis()\n",
    "plt.savefig('ipmo_density_plot_01')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "xy = np.vstack([x_bd,y_bd])\n",
    "z = gaussian_kde(xy)(xy)\n",
    "\n",
    "idx = z.argsort()\n",
    "x, y, z = x_bd[idx], y_bd[idx], z[idx]\n",
    "\n",
    "fig, ax = plt.subplots()\n",
    "fig.set_size_inches(15,9)\n",
    "ax.set_facecolor((64/255, 3/255, 80/255))\n",
    "im = ax.scatter(x, y, c=z, s=10000, edgecolor='')\n",
    "cb = plt.colorbar(im,fraction=0.046)\n",
    "cb.set_label('Surface Density per Square Degree',size=20)\n",
    "cb.ax.set_yticklabels([int(0.05*(bd_density/circ_area)),int(0.10*(bd_density/circ_area)),\n",
    "                       int(0.15*(bd_density/circ_area)),int(0.20*(bd_density/circ_area)),\n",
    "                       int(0.25*(bd_density/circ_area)),int(0.30*(bd_density/circ_area))],size=16)\n",
    "#ax.scatter(x, y, color='k',s=1)\n",
    "ax.scatter(84.05224370308734,-1.1913836528004638, color='r', marker ='*', s=120)\n",
    "plt.title('BD Candidate Density',fontsize=36)\n",
    "plt.xlabel('RA', fontsize=28)\n",
    "plt.ylabel('Dec', fontsize=28)\n",
    "plt.gca().invert_xaxis()\n",
    "plt.savefig('bd_density_plot_01')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "xy = np.vstack([x_yso,y_yso])\n",
    "z = gaussian_kde(xy)(xy)\n",
    "\n",
    "idx = z.argsort()\n",
    "x, y, z = x_yso[idx], y_yso[idx], z[idx]\n",
    "norm = cm.colors.Normalize(vmax=abs(z).max(), vmin=-abs(z).max())\n",
    "fig, ax = plt.subplots()\n",
    "fig.set_size_inches(15,9)\n",
    "ax.set_facecolor((64/255, 3/255, 80/255))\n",
    "im = ax.scatter(x, y, c=z, s=10000, edgecolor='')\n",
    "cb = plt.colorbar(im,fraction=0.046)\n",
    "cb.set_label('Surface Density per Square Degree',size=20)\n",
    "cb.ax.set_yticklabels([int(0.05*(yso_density/circ_area)),int(0.10*(yso_density/circ_area)),\n",
    "                       int(0.15*(yso_density/circ_area)),int(0.20*(yso_density/circ_area)),\n",
    "                       int(0.25*(yso_density/circ_area)),int(0.30*(yso_density/circ_area))],size=16)\n",
    "#ax.scatter(x, y, color='k',s=1)\n",
    "ax.scatter(84.05224370308734,-1.1913836528004638, color='r', marker ='*', s=120)\n",
    "plt.title('YSO Candidate Density',fontsize=36)\n",
    "plt.xlabel('RA', fontsize=28)\n",
    "plt.ylabel('Dec', fontsize=28)\n",
    "plt.gca().invert_xaxis()\n",
    "plt.savefig('yso_density_plot_01')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "def coord_convert(file,output_name):\n",
    "    info = fits.open(file)\n",
    "    tbl = info[1].data\n",
    "    \n",
    "    RA_ICRS = []\n",
    "    Dec_ICRS = []\n",
    "    \n",
    "    for x in range(len(tbl)):\n",
    "        ra_h = int(str(tbl[x][1])[2:3])\n",
    "        ra_m = int(str(tbl[x][1])[3:5])\n",
    "        ra_s = float(str(tbl[x][1])[5:9])/100\n",
    "        dec_sign = np.sign(float(str(tbl[x][1])[9:17]))\n",
    "        dec_h = int(str(tbl[x][1])[10:12])\n",
    "        dec_m = int(str(tbl[x][1])[12:14])\n",
    "        dec_s = float(str(tbl[x][1])[14:17])/10\n",
    "        \n",
    "        ra = ra_h*15 + ra_m*15/60 + ra_s*15/3600\n",
    "        dec = dec_sign*(dec_h + dec_m/60 + dec_s/3600)\n",
    "        \n",
    "        RA_ICRS.append(ra)\n",
    "        Dec_ICRS.append(dec)\n",
    "        \n",
    "    col3a = fits.Column(name='RA_ICRS',format='D',unit='deg',array=np.array(RA_ICRS))\n",
    "    col4a = fits.Column(name='Dec_ICRS',format='D',unit='deg',array=np.array(Dec_ICRS))\n",
    "    \n",
    "    cols = fits.ColDefs([col3a,col4a])\n",
    "    \n",
    "    output_fits = output_name + '.fits'\n",
    "    \n",
    "    data = fits.BinTableHDU.from_columns(cols)\n",
    "    data.writeto(output_fits)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "coord_convert('J_A+A_485_931_tablea15.dat.fits','caballero15')"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.6.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}