简单学量化pandas的应用14nlargest和nsmallest函数

今天我们学习nlargest和nsmallest函数，这是pandas中的两个排序并取前多少名的函数。

我们先来看一下传统方式。

首先，获取数据，还是用上次讲函数用过的股票数据。

df = pro.daily(ts_code='000333.SZ', start_date='20240201', end_date='20240317')
df = df[['trade_date','open','close']]
df

返回（部分截图）：

#先按close由高到低排序，然后取前5条。df.sort_values(by='close', ascending=False).head(5)

返回：

用nlargest函数是这样的：

df.nlargest(5, 'close')#取close最大的5条。

返回：

一样的结果，但代码简单易懂。

同理，可以用nsmallest取某一列数值最小的多少条：

df.nsmallest(5,'open')#取open最小的5条。

返回：

通过以上的例子，两个会用nlargest和nsmallest这两个函数了吧？