#ylabel='Time (10 s)') plot_lib.image_plot(fig, ax[0], ins.data_ms[:, 0, :, 0], freq_array=ins.freq_array[0], cmap=cm.coolwarm, aspect=aspect, cbar_label='Deviation ($\hat{\sigma}$)', vmin=-5, vmax=5, ylabel='Time (10 s)') fig_mf, ax_mf = plt.subplots(figsize=(16, 9), ncols=2) ins.data.mask[:, 0, :82, 0] = True ins.data.mask[:, 0, -21:, 0] = True ins.data_ms = ins.mean_subtract(order=1) mf = MF(ins, sig_thresh=5, N_thresh=0, shape_dict={ 'TV4': [1.74e8, 1.82e8], 'TV5': [1.82e8, 1.9e8], 'TV6': [1.9e8, 1.98e8] }) mf.apply_match_test(apply_N_thresh=False, order=1) plot_lib.image_plot(fig_mf, ax_mf[0], ins.data[:, 0, :, 0], freq_array=ins.freq_array[0], cbar_label='Amplitude (UNCALIB)',
read_paths = { 'data': arr, 'Nbls': '%s/arrs/%s_%s_INS_Nbls.npym' % (args.inpath, obs, args.flag_choice), 'freq_array': '%s/metadata/%s_freq_array.npy' % (args.inpath, obs), 'pols': '%s/metadata/%s_pols.npy' % (args.inpath, obs), 'vis_units': '%s/metadata/%s_vis_units.npy' % (args.inpath, obs) } ins = INS(obs=obs, outpath=args.outpath, flag_choice=args.flag_choice, read_paths=read_paths) # ins.data[:, :, :82] = np.ma.masked # ins.data[:, :, -21:] = np.ma.masked ins.data_ms = ins.mean_subtract() ins.counts, ins.bins, ins.sig_thresh = ins.hist_make() cp.INS_plot(ins, **ms_plot_kwargs) mf = MF(ins, shape_dict=shape_dict, point=args.point, streak=args.streak, **mf_kwargs) for test in args.tests: getattr(mf, 'apply_%s_test' % test)(args.order) ins.save() cp.MF_plot(mf, **ms_plot_kwargs)
'vmin': -5, 'vmax': 5, 'mask_color': 'black' }] for i in range(2): for k, obj in zip([0, 2], [raw_ins, OR_ins]): pl.image_plot(fig, ax[k, i], getattr(obj, attr[i])[:, 0, :, 0], cbar_label=cbar_label[i], freq_array=obj.freq_array[0], **kwargs[i]) raw_ins.data[:, 0, :82] = np.ma.masked raw_ins.data[:, 0, -21:] = np.ma.masked mf = MF(raw_ins, sig_thresh=5, shape_dict=shape_dict, N_thresh=20) raw_ins.data_ms = raw_ins.mean_subtract(order=1) mf.apply_match_test(apply_N_thresh=True, order=1) raw_ins.data_ms = raw_ins.mean_subtract() for i in range(2): pl.image_plot(fig, ax[1, i], getattr(raw_ins, attr[i])[:, 0, :, 0], cbar_label=cbar_label[i], freq_array=raw_ins.freq_array[0], **kwargs[i]) fig.savefig('%s/%s_flag_comp.png' % (figpath, obs)) raw_ins.save() plt.close(fig)