dfk_summary.py

import numpy as np
import plot_simple as ps
import plot_allZ2 as pa2
import sys

if len(sys.argv) == 4:
    exclude = []
    for i in range(6):
        d, e = np.loadtxt('P{}_GBU.dat'.format(i+1),unpack=True)
        idx = (np.where(e != 0)[0]+1).tolist()
        exclude.append(idx)
else:
    exclude = [[5,34],
               [1,2,35],
               [59],
               [2,8],
               [1,2,3,27,28,29,5],
               [35,36,38]]

print exclude
pre = sys.argv[1]

try:
    offset = int(sys.argv[2])
except IndexError:
    offset = 0

# ps.all_maps('{}_MLWA_map.pdf'.format(pre),col=6+offset,inputsuffix='allz2.dat',label='Mean Light Weighted Age [Gyr]', minval=0, maxval=12.5,binned=True,exclude=exclude)
# ps.all_maps('{}_MMWA_map.pdf'.format(pre),col=5+offset,inputsuffix='allz2.dat',label='Mean Mass Weighted Age [Gyr]', minval=0, maxval=12.5,binned=True,exclude=exclude)
# ps.all_maps('{}_MLWZ_map.pdf'.format(pre),col=8+offset,inputsuffix='allz2.dat',label='Mean Light Weighted Metallicity [Z_sol]', minval=-1.5, maxval=0.5,binned=True,log=True,exclude=exclude)
# ps.all_maps('{}_V_map.pdf'.format(pre),col=9+offset,inputsuffix='allz2.dat',label='Velocity [km/s]', minval=298, maxval=758,binned=True,cmap='bwr',exclude=exclude)
# ps.all_maps('{}_TauV_map.pdf'.format(pre),col=10+offset,inputsuffix='allz2.dat',label=r'$\tau_V$', minval=0, maxval=5,binned=True,exclude=exclude)

pa2.dfk_batch(sys.argv[1],offset=offset,exclude=exclude)