preprocess_pipe.py 文件源码

def get_notwr_data(idx, spec_dir, filenames):
    spec_path = spec_dir + filenames[idx]
    SpecID = spec_path[6:21]

    hdulist = fits.open(spec_path)

    # Get wavelength data
    wav_rest = 10 ** hdulist[1].data['loglam']

    # Get flux density, in this case erg/cm^2/s/Angstrom.
    fwav = hdulist[1].data['flux']

    crop_range = [4686 - 250, 4686 + 250]
    wav_rest, fwav = crop_data(wav_rest, fwav, crop_range)

    hdulist.close()

    return wav_rest, fwav, SpecID