查找具有多列最大和的行

Question

我用代码创建DataFrame

import pandas as pd

data = [
    ["id_1",6,7,9],
    ["id_2",9,7,1],
    ["id_3",6,7,10],
    ["id_4",9,5,10]
]
df = pd.DataFrame(data, columns = ['Student Id', 'Math', 'Physical', 'Chemistry'])

我们如何找到具有max（数学+物理+化学）的行，结果= id_4

谢谢您！

Answer 1

您可以将“学生ID”列设置为索引，然后计算最大和。

df = df.set_index('Student Id')
print(df.sum(axis=1).idxmax())