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]

我所做的是首先獲取學生行的索引，然后簡單地返回相同索引的標記。