Python - How I pass a vector in where condition

Asked Apr 06 '18 at 01:16

Active Apr 06 '18 at 01:16

Viewed 31 times

I am new to Python Data Science. Did work in R and exploring Python for ML. I used pass vector in R where condions but I am not sure how I do that in Python.

raw_data1 = {'age1': [23,45,21],'age2': [10,20,50], 'age3':[44,55,77]}
df1 = pd.DataFrame(raw_data1, columns = ['age1','age2','age3'])

raw_data2 = {'age1': [23,45]}
df2 = pd.DataFrame(raw_data2, columns = ['age1'])

Here I like to query df1 based on df2-age1 vector. I tried below but didn't work

df3 = df1.iloc[np.where(df1.age1 == df2['age1'])]

Thanks for your help

asked Apr 06 '18 at 01:16

Murali

1

`df1[df1.age1.isin(df2.age1)]` – cs95 Apr 06 '18 at 01:18
Try adding parenthesis between each part of your condition: `(df1.age1) == (df2.age1)` – edthrn Apr 06 '18 at 01:21
@cᴏʟᴅsᴘᴇᴇᴅ Thank you. It is working. BTW, how I deal with if my varaibles are in text (not numbers). I tried with same syntax but didn't work. – Murali Apr 06 '18 at 01:57
@COLDSPEED It is working for both numbers and strings. Thank you – Murali Apr 06 '18 at 03:41

Python - How I pass a vector in where condition

0 Answers0