def pick_stock(stock, species):
    """Return the stock for single species.

    Parameters
    ----------
    stock : np.array
    species : int
        0 for prey, 1 for predator.

    Returns
    -------
    np.array

    """
    return stock[:, :, species]