def extract_feature(array, n_samples=1):
    """Extracts the feature column from an array.

    Note that this method expects the first column to the the row identifier
    and the second column to be the feature.

    Args:
        array (np.array): Numpy array to be extracted.
        n_samples (int): Maximum number of samples.

    Returns:
        The extracted feature column out of an array

    """

    # Parses the data with n_samples and using the last column as the feature column
    feature = array[:n_samples, -1]

    return feature