def figure_init(): figure(figsize=(6.5,8.5)) pretty.big_tick_labels(16) pretty.long_ticks() pretty.set_pretty() cc = pretty.get_colors() return 0
#!/usr/bin/env python from pylab import * import py_read_output as r import py_plot_output as pl import py_plot_util as util import os, sys from constants import * import numpy as np import pretty import mynormalize import sdss_sub_data as sub # from colorbar import DraggableColorbar as drbar cc = pretty.get_colors() class selection: ''' a class which contains a number of Boolean arrays used for easy selection of subsamples in the SDSS data ''' def __init__(self, data, datatype="sdss"): if datatype == "sdss": self.nonbal = (data["bal_flag"] == 0) self.mgbal = (data["bal_flag"] == 2) #self.z = (data["z"] > redshift_lims[0]) * (data["z"] < redshift_lims[1]) * (data["z"] < 2) self.z = (data["z"] < 1) self.has_o3 = (data["ew_o3"] > 0)