pandas 获取最大值对应列名

import numpy as np
import pandas as pd
df = pd.DataFrame({
    'id':['x', 'y', 'z'],
    'a':[3,4,2],
    'b':[0,5,7],
    'c':[1,0,1]}
)
df

利用idxmax，获取最大值对应的行索引。而通过设置axis=1，可返回最大值对应的列名。

df['最高分']=df.iloc[:,1:].apply(max,axis=1)
df['最高分科目']=df.iloc[:,1:].idxmax(axis=1)
df
#也可以先将第一列设置为行索引，以便后续相应操作。