有沒有辦法將值列表傳遞給 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