def tta_precip(): import Windprof2 as wp print "BBY-tta | BBY-notta | CZD-tta | CZD-notta | FRS-tta | FRS-notta " o = "{:11.1f} {:11.1f} {:11.1f} {:11.1f} {:11.1f} {:11.1f}" for case in range(8, 15): bby, czd, frs, _ = get_data(str(case)) tta_times = wp.get_tta_times(case=str(case), continuous=True) if len(tta_times) > 0: tta_idx_bby = (bby.index >= tta_times[0]) & (bby.index <= tta_times[-1]) tta_idx_czd = (czd.index >= tta_times[0]) & (czd.index <= tta_times[-1]) tta_idx_frs = (frs.index >= tta_times[0]) & (frs.index <= tta_times[-1]) bby_tta = bby.precip[tta_idx_bby].sum() bby_notta = bby.precip[~tta_idx_bby].sum() czd_tta = czd.precip[tta_idx_czd].sum() czd_notta = czd.precip[~tta_idx_czd].sum() frs_tta = frs.precip[tta_idx_frs].sum() frs_notta = frs.precip[~tta_idx_frs].sum() else: bby_tta = -999 bby_notta = bby.precip.sum() czd_tta = -999 czd_notta = czd.precip.sum() frs_tta = -999 frs_notta = frs.precip.sum() print o.format(bby_tta, bby_notta, czd_tta, czd_notta, frs_tta, frs_notta)