如何使用 pandas idxmax 函数获取值？

Question

当我们将 idxmax() 函数应用于 pandas DataFrame 时，它​​会返回表的索引和列。是否也有熊猫方法可以打印出值？

例如：

np.random.seed(seed=42)
df = pd.DataFrame(data=np.random.randint(0, 10, (10,10)),
           columns=["Test{}".format(x) for x in range(1, 10 + 1)],
            index=["Student{}".format(x) for x in range(1, 10 + 1)])

df.idxmax(axis =1)

它返回如下内容：

Student1      Test6
Student2      Test2
Student3      Test3
Student4      Test1
Student5      Test3
Student6     Test10
Student7      Test4
Student8      Test6
Student9      Test2
Student10     Test4

我也可以使用 pandas 为这些索引打印值吗？

Answer 1

将DataFrame.agg 与idxmax 和max 一起使用：

df = df.agg(['idxmax','max'],axis=1)
print (df)
           idxmax max
Student1    Test6   9
Student2    Test2   7
Student3    Test3   9
Student4    Test1   8
Student5    Test3   9
Student6   Test10   9
Student7    Test4   9
Student8    Test6   9
Student9    Test2   8
Student10   Test4   9

---

df1 = df.agg(['idxmax','max'],axis=1).astype({'max':int}).nlargest(5, 'max')
print (df1)

          idxmax  max
Student1   Test6    9
Student3   Test3    9
Student5   Test3    9
Student6  Test10    9
Student7   Test4    9