def get_mag(time_steps, total_flux, wave, trans, data): mag = assign_fluxes.calculate_AB_mag(time_steps, data[1:, 0], total_flux, wave, trans) return mag
def get_mag(total_flux, wave, trans, data): mag = N.zeros(len(total_flux)) for n in range(len(total_flux)): mag[n] = assign_fluxes.calculate_AB_mag(data[1:,0], total_flux[n,:], wave, trans) return mag