Compare a list/array to a dataframe column?

Question

I have a list or array contains values[apple,items] and i want a resultant df with where column('c') has the same list/array values.

Suppose in below eg: A can be a list/array/str and resultant df is based on the values in A compared with df 'c' column

eg:

   A = [apple,items]

df

                        a         b        c

                0       10        13      items

                1       9         12      testcase

                2       8         11      apple

                3       7         10      apple

                4       6         9       test

                5       5         8       items

Final df

                        a         b        c

                0       10        13      items

                1       8         11      apple

                2       7         10      apple

                3       5         8       items

I tried with boolean indexing but couldn't succeced

pandas series.isin()

G. Anderson
– G. Anderson

2019-09-17 18:26:58 +00:00
Commented Sep 17, 2019 at 18:26 — G. Anderson
– G. Anderson, Commented Sep 17, 2019 at 18:26

Celius Stingher · Accepted Answer · 2019-09-17 18:29:23Z

1

You should be able to work it out with this:

df['Status'] = df['c'].isin(A)

This should add a column named Status that will be True when the row for column C is apple or items, False otherwise.

answered Sep 17, 2019 at 18:29

Celius Stingher

18.4k6 gold badges26 silver badges54 bronze badges

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

Comments

rahlf23 · Accepted Answer · 2019-09-17 18:33:57Z

1

You want to use pd.Series.isin() and pass the results as a boolean mask to filter your original dataframe:

A = ['apples','items']

df_final = df[df['c'].isin(A)]

answered Sep 17, 2019 at 18:33

rahlf23

9,0494 gold badges30 silver badges57 bronze badges

Comments

Matias Guerrero Irarrázaval · Accepted Answer · 2019-09-17 18:48:54Z

1

You create a boolean:

selection = df['c'].isin(A)

and then get your final df from it:

Final_df = df[selection]

answered Sep 17, 2019 at 18:48

Matias Guerrero Irarrázaval

216 bronze badges

Collectives™ on Stack Overflow

Compare a list/array to a dataframe column?

3 Answers 3

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related