Function 使用 str 索引获取 dataframe 的行值

Question

样本df：

Student  Marks
Avery     70
Joe       80
John      75
Jordan    90

当学生姓名通过时，我想使用如下 function 来返回分数。

def get_marks(student):
    return *something*

预期 Output：get_marks('Joe') ==> 80

Answer 1

我认为以下可能有效。

def get_marks(student):
    p = df.index[df['Student'] == student].tolist()
    p = p[0]
    return df['Marks'][p]

我所做的是首先获取学生行的索引，然后简单地返回相同索引的标记。