有没有一种方法可以将值列表传递给pandas数据帧的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([300400](]

结果

loan_purpose    loan_amount
1   B   300
2   B   400