FROM PhotoObj AS p JOIN SpecObj AS s ON s.bestobjid = p.objid WHERE (s.BOSS_TARGET1 & 1) != 0 and s.bossprimary = 1 and ZWARNING_NOQSO = 0 and s.z between 0.002 and 0.5 and s.plateID >= 10324 ''' # redoanlaod with new query from CSV import CSV from calc_kcor import calc_kcor from astropy.cosmology import Planck15 as cosmo import numpy as np import matplotlib.pyplot as pl SDSS = CSV('data_all_2') SDSS_data = SDSS.read_all() # -1 because of column titles print(SDSS.row_count_data) calculated = np.zeros((SDSS.row_count_data, 3)) #add area print('read data') for index, row in enumerate(SDSS_data): if index % 1000 == 0: pass z, zErr_noqso, cModelMag_r, cModelMagErr_r, cModelMag_u, cModelMagErr_u, petroRad_r, petroRadErr_r, modelMag_r, modelMag_u, petroMag_r, petroMagErr_r, petroMagErr_u, petroMag_u, Pr90 = row dist = cosmo.luminosity_distance(z).value # in Mpc k_corr = calc_kcor('r', z, 'u - r', (petroMag_u - petroMag_r))