如何提取数据框中某列的前两个值的ID？

Question

我有一个数据框，从中需要一个特定列的前两个值的ID（索引值）。

示例

two_highest_degree_id=karate.nlargest(2,'degree')

空手道是我的数据框，我需要列度的前两个值的索引值。上面的脚本给了我以下结果：

      club     degree  cluster_coef
33  Officer      17      0.110294
0    Mr. Hi      16      0.150000

如何为输出分配索引值33和0

Answer 1

import pandas as pd
df = pd.DataFrame([10,9,8,7,6,50], columns = ['degree'])

df.nlargest(2,'degree').index
#op
Int64Index([5, 0], dtype='int64')