コード例 #1
0
def process(file_0, file_1):

    kwargs_relint = {'smooth': 20, 'scale': None, 'decmax': -55}

    relint_0 = anisotropy.get_map(files=file_0, name='relint', **kwargs_relint)
    relint_1 = anisotropy.get_map(files=file_1, name='relint', **kwargs_relint)

    relerr_0 = anisotropy.get_map(files=file_0, name='relerr', **kwargs_relint)
    relerr_1 = anisotropy.get_map(files=file_1, name='relerr', **kwargs_relint)

    ri_0, ri_err_0, ra, ra_err = anisotropy.get_proj_relint(relint_0,
                                                            relerr_0,
                                                            n_bins=24)
    ri_1, ri_err_1, ra, ra_err = anisotropy.get_proj_relint(relint_1,
                                                            relerr_1,
                                                            n_bins=24)

    ks_statistic, pval = ks_2samp(ri_0, ri_1)
    chi2 = calc_red_chi2(ri_0, ri_1, ri_err_0, ri_err_1)

    return pval, chi2
コード例 #2
0
        raise ValueError('Input files must be specified')
    elif len(args.infiles_sample_0) != len(args.infiles_sample_1):
        raise ValueError('Both samples of input files must be the same length')

    if args.outfile is None:
        raise ValueError('Outfile must be specified')
    else:
        comp.check_output_dir(args.outfile)

    data_dict = {'ks_statistic': [], 'pval': []}
    # Read in all the input maps
    kwargs_relint = {'smooth': 20, 'scale': None, 'decmax': -55}
    for file_0, file_1 in zip(args.infiles_sample_0, args.infiles_sample_1):

        relint_0 = anisotropy.get_map(files=file_0,
                                      name='relint',
                                      **kwargs_relint)
        relint_1 = anisotropy.get_map(files=file_1,
                                      name='relint',
                                      **kwargs_relint)

        ri_0, ra, ra_err = anisotropy.get_proj_relint(relint_0, n_bins=100)
        ri_1, ra, ra_err = anisotropy.get_proj_relint(relint_1, n_bins=100)

        print('Comparing:')
        print('ri_0 = {}'.format(ri_0))
        print('ri_1 = {}\n'.format(ri_1))
        ks_statistic, pval = ks_2samp(ri_0, ri_1)
        print('ks_statistic = {}'.format(ks_statistic))
        print('pval = {}\n\n'.format(pval))
        data_dict['ks_statistic'].append(ks_statistic)