如果参数大于长度，如何使 nlargest 函数给出错误

Question

我希望函数在大于系列长度nlargest时给出错误。n有什么办法可以做到这一点pd.Series([1,2,3,4,5]).nlargest(10)正如你所看到的，系列的长度小于 10，所以我想要一个错误。我应该怎么办

Answer 1

也许只需为其编写一个函数：

def custom_nlargest(series, n):

    if n > len(series):

        raise ValueError("n is greater than series length!")

    else:

        return series.nlargest(n)

Answer 2

将其包装在函数中怎么样：

def my_nlargest(series, n):
    if len(series) < n: return 'error message'
    return series.nlargest(n)