""" TABLE_DATA = [] TABLE_DATA.append( ['Data set', 'slope', 'intercept', 'r', 'p-value']) for s in range(Globals.n_sets): TABLE_DATA.append([ str(Globals.dataset_names[s]), str(round(lt_params[s][0], 7)), str(round(lt_params[s][1], 5)), str(round(lt_params[s][2], 5)), str(lt_params[s][3]) ]) TABLE_DATA = tuple(TABLE_DATA) table_instance = SingleTable(TABLE_DATA, "Linear trend fit") table_instance.justify_column = 'right' print "" print(table_instance.table) except: #if terminaltables is not installed for s in range(Globals.n_sets): print "" print_message( '\t/' + str(s) + " Data set:" + str(Globals.dataset_names[s]), 3, 32) print "\tslope:", lt_params[s][0] print "\tintercept:", lt_params[s][1] print "\tr:", lt_params[s][2] print "\tp-value:", lt_params[s][3]
def load_multiset_data(): """ """ #multiset Globals.n_sets = 0 print_message("\nReading multiset data\n", index=4, color=34) #count data files passed as CLI arguments Globals.times, Globals.rvs, Globals.rv_errs = [],[],[] Globals.dataset_names, DATA = [], [] Globals.mean_err = [] try: from terminaltables import AsciiTable, DoubleTable, SingleTable ttable = True except: ttable = False s = 0 for arg in sys.argv[1:]: if arg[0] != '-': #read data try: in_data = mgls_io.read_file(arg, ' ') if not ttable: print_message("\t" + str(s) + "/ Read data file " + arg, index=3, color=32) except IOError: print_message("\tData file could not be read", 5, 31) sys.exit() #assign data vectors time, rv, rv_err = mgls_io.get_data_vectors(in_data, Globals.col) if Globals.km2m: rv *= 1000.0 rv_err *= 1000.0 Globals.dataset_names.append( str(s) + '/ ' + arg.split('/')[-1]) #file name. Not full path Globals.times.append(time) mean_rv = np.mean(rv) mean_rv_err = np.mean(rv_err) Globals.mean_err.append(mean_rv_err) Globals.rvs.append(rv) Globals.rv_errs.append(rv_err) inv_sigma2 = 1.0/(rv_err**2) logL_0 = -0.5*(np.sum(((rv-mean_rv)**2)*inv_sigma2 + np.log(2.0*np.pi) + np.log(1.0/inv_sigma2)) ) Globals.n_sets += 1 #print info summ = 0.0 separation = [] for i in range(len(time)-1): summ += time[i+1] - time[i] separation.append(time[i+1] - time[i]) if ttable: DATA.append([len(time), time[-1] - time[0], summ/len(time), logL_0]) else: if not Globals.inhibit_msg: print "\t-----------------Summary of data--------------------" print "\tData points:", len(time) print "\tTime span:", time[-1] - time[0] print "\tMean sep.", summ/len(time) print "\tlogL null model (data, no jitter):", logL_0 print "\t----------------------------------------------------" #count dataset s += 1 #print ttable (if appliable) if ttable: TABLE_DATA = [] TABLE_DATA.append(['Data set name (full path not shown)', 'Data points', 'Timespan', 'Mean sep.', 'logL (data)']) for s in range(Globals.n_sets): TABLE_DATA.append([str(Globals.dataset_names[s]), str(DATA[s][0]), str(round(DATA[s][1],5)), str(round(DATA[s][2],5)), str(round(DATA[s][3],5))]) TABLE_DATA = tuple(TABLE_DATA) table_instance = SingleTable(TABLE_DATA, "Dataset summary") table_instance.justify_column = 'right' print(table_instance.table) Globals.len_sets = [len(Globals.times[i]) for i in range(len(Globals.times))] Globals.times_seq, Globals.rvs_seq, Globals.rv_errs_seq = [], [], [] for i in range(len(Globals.times)): Globals.times_seq.extend(Globals.times[i]) Globals.rvs_seq.extend(Globals.rvs[i]) Globals.rv_errs_seq.extend(Globals.rv_errs[i]) Globals.times_seq = np.array(Globals.times_seq) Globals.rvs_seq = np.array(Globals.rvs_seq) Globals.rv_errs_seq = np.array(Globals.rv_errs_seq)