有没有办法将值列表传递给 pandas dataframe 的 df.loc 方法？

Question

我正在尝试使用从用户那里收到的值过滤数据框。 我面临的问题是，如果输入有多个值，我找不到使用 loc 方法的方法。 假设贷款金额是 300、400 和 600。理想情况下，我想传递一个值列表。 我怎样才能做到这一点？ 请参阅下面的代码以了解过滤

mortgage_df = df.loc[(df['loan_purpose'] == loan_purpose) & (df['loan_amount'] == loan_amount)]

Answer 1

样本输入

df = pd.DataFrame({
    "loan_purpose": ['A', 'B', 'B', 'D'],
    'loan_amount' : [200, 300, 400, 500]
})

df.loc[(df.loan_purpose == 'B') & df.loan_amount.isin([300, 400])]

结果

loan_purpose    loan_amount
1   B   300
2   B   400