import numpy as np
avgdat=np.genfromtxt('avg_spec_data',skip_header=1)
sa_dat=np.genfromtxt('report_spectrum',skip_header=1)
freq=avgdat[:,0]
sa_t_avg=sa_dat[:,1]
sa_t_stderr=sa_dat[:,2]
empty_t_avg=avgdat[:,5]
empty_t_stderr=avgdat[:,6]

sa_r_avg=sa_dat[:,3]
sa_r_stderr=sa_dat[:,4]/np.sqrt(sa_dat[:,5])
empty_r_avg=avgdat[:,7]
empty_r_stderr=avgdat[:,8]
with open('extract_data','w') as f:
    f.write('{:^10} {:^10} {:^10} {:^10} {:^10} {:^10} {:^10} {:^10} {:^10}\n'.format('freq','avg_T','stderr_T','empty_T','stderr_empty_T','avg_R','stderr_R','empty_R','stderr_empty_R'))
    for i in range(len(freq)):
            f.write('{:^10} {:^10.3f} {:^10.3f} {:^10.3f} {:^10.5f} {:^10.3f} {:^10.3f} {:^10.3f} {:^10.5f} \n'.format(int(freq[i]),sa_t_avg[i],sa_t_stderr[i],empty_t_avg[i],empty_t_stderr[i],sa_r_avg[i],sa_r_stderr[i],empty_r_avg[i],empty_r_stderr[i]))

with open('R_atom','w') as f:
    for i in range(len(freq)):
            f.write('{:^10} {:^10.3f} {:^10.5f}  \n'.format(int(freq[i])-360,sa_r_avg[i],sa_r_stderr[i]))

with open('R_woatom','w') as f:
    for i in range(len(freq)):
            f.write('{:^10} {:^10.3f} {:^10.5f}  \n'.format(int(freq[i])-360,empty_r_avg[i],empty_r_stderr[i]))

