Python Subset dataframe with another dataframe

Question

I'm transitioning from R to Python and trying to subset a dataframe with a field in another dataframe. What would be the Python Equivalent for this R code:

final_solution <- subset(df1, item %in% df2$item)

Thanks

Are you using pandas?

Alex Riley
– Alex Riley

2015-12-19 18:09:47 +00:00
Commented Dec 19, 2015 at 18:09 — Alex Riley
– Alex Riley, Commented Dec 19, 2015 at 18:09
yes i'm using pandas

BlackHat
– BlackHat

2015-12-19 19:40:58 +00:00
Commented Dec 19, 2015 at 19:40 — BlackHat
– BlackHat, Commented Dec 19, 2015 at 19:40

akrun · Accepted Answer · 2015-12-19 18:25:46Z

2

We can try

df1[(df1.item).isin(df2.item)]

Using a reproducible example (with pandas)

import pandas as pd
df1 = pd.DataFrame({'item' :  [1, 2, 3, 4],
                           'fruit' : ['mango', 'apple', 'banana', 'mango']})

df2 = pd.DataFrame({'item' : [1, 2]})

print(df1[(df1.item).isin(df2.item)])

gives the output

#    fruit  item
#0  mango     1
#1  apple     2

answered Dec 19, 2015 at 18:25

akrun

891k38 gold badges590 silver badges700 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

BlackHat Over a year ago

Exactly what I was looking for.

Padraic Cunningham · Accepted Answer · 2015-12-19 18:58:44Z

2

Presuming you are using pandas, you could merge:

pd.merge(df1, df2, how='inner', on=['df1_col', 'df2_col'])

answered Dec 19, 2015 at 18:58

Padraic Cunningham

181k30 gold badges264 silver badges327 bronze badges

Collectives™ on Stack Overflow

Python Subset dataframe with another dataframe

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related