Source code for oospectro.io

import numpy as np


def load_spectrum(spectrum_path, lambda_min=100, lambda_max=1000):
    """
    Load a spectrum file.

    Parameters
    ----------
    spectrum_path : string
        File path.
    lambda_min : scalar, optional
        Cut the data at this minimum wavelength in nm.
    lambda_max : scalar, optional
        Cut the data at this maximum wavelength in nm.

    Returns
    -------
    values : arrays
        (lamdbas, intensities)
    """
    data = np.loadtxt(spectrum_path, delimiter=',')
    lambdas, intensities = np.column_stack(data)

    mask = (lambdas > lambda_min) & (lambdas < lambda_max)
    return lambdas[mask], intensities[mask]