如果参数大于长度，如何使 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&nbsp;my_nlargest(series,&nbsp;n): &nbsp;&nbsp;&nbsp;&nbsp;if&nbsp;len(series)&nbsp;<&nbsp;n:&nbsp;return&nbsp;'error&nbsp;message' &nbsp;&nbsp;&nbsp;&nbsp;return&nbsp;series.nlargest(n)