Use a list of values to select rows from a pandas dataframe

Question

Let's say I have the following pandas dataframe:

>df = DataFrame({'A' : [5,6,3,4], 'B' : [1,2,3, 5]})

>Df

  A B

0 5 1

1 6 2

2 3 3

3 4 5

I can subset based on a specific value:

x = df[df['A'] == 3]

x

  A B

2 3 3

But how can I subset based on a list of values? - something like this:

list_of_values = [3,6]

y = df[df['A'] in list_of_values]

Answer 1

For this problem, you can use isin(). This function works the same as the SQL query. It searches for the given value in the dataframe to selects the entire row.

For example:

>df = DataFrame({'A' : [5,6,3,4], 'B' : [1,2,3, 5]})

>Df

   A B

0 5 1

1 6 2

2 3 3

3 4 5

>df[df['A'].isin([3, 6])]

   A B

1 6 2  

2 3 3

Hope this answer helps.

Learn Pandas Tutorial here.