Pandas nlargest() 函数：返回最大值的用法详解

在 Pandas 中，nlargest() 函数用于返回 DataFrame 或 Series 中最大的 n 个值。它接收一个参数 n，表示要返回的最大值的数量。

1. DataFrame 的用法

import pandas as pd

data = {'Name': ['Tom', 'Nick', 'John', 'Kevin'],
        'Age': [25, 30, 35, 40],
        'Salary': [50000, 60000, 70000, 80000]}

df = pd.DataFrame(data)

# 返回 Salary 列中最大的 2 个值所在的行
result = df.nlargest(2, 'Salary')
print(result)

输出：

   Name  Age  Salary
3  Kevin   40   80000
2   John   35   70000

2. Series 的用法

import pandas as pd

data = pd.Series([5, 2, 8, 1, 3])

# 返回 Series 中最大的 3 个值
result = data.nlargest(3)
print(result)

输出：

2    8
0    5
4    3
dtype: int64

可以看到，nlargest() 函数返回了 DataFrame 或 Series 中最大的 n 个值，并按照降序排列。对于 DataFrame，可以指定要基于哪一列进行排序。

总结

nlargest() 函数是 Pandas 中一个非常实用的函数，可以方便地从 DataFrame 或 Series 中获取最大的值，并根据需要进行排序。在数据分析和处理中，它可以帮助我们快速找到关键数据点。