通过使用 pandas 在循环中比较列和变量来为列分配值

Question

我有两列，一列是学生姓名，另一列是空的。 我有一个包含已通过学生姓名的列表。 所以在 for 循环中，我将从列表中获取每个名称并将其与第一列值进行比较。 如果在该列中找到名称，则在与该名称对应的第二列中添加“通过”。

必须通过列表中的每个值 go 并在第二列中填写 PASS/FAIL。 但问题是，我不知道如何在 pandas 条件下传递变量名。 如果我直接传递变量的值，它就可以工作。 有人可以建议，如何直接传递变量名而不是值。

    df.loc[df['student_name'] == "John", 'status'] = "Pass"
    

我想用循环中的变量名（passed_students）替换这个名字“John”，就像：

      df.loc[df['student_name'] == passed_students, 'status'] = "Pass"

但它正在抛出“KeyError：'student_name'”

Answer 1

使用.isin() function 对操作进行矢量化。

import pandas as pd
import string
import numpy as np

## create dummy data
df = pd.DataFrame({"students":list(string.ascii_lowercase[:26]),"status":"Failed"})
passed_students = list(string.ascii_lowercase[:6])

print(df.head())
##look for students who have passed and assign "Passed" status.
df.loc[df.students.isin(passed_students),"status"] = "Passed"
print(df.head())