Use a list of values to select rows from a Pandas DataFrame



To select the rows from a Pandas DataFrame based on input values, we can use the isin() method.

Steps

  • Create a two-dimensional, size-mutable, potentially heterogeneous tabular data, df.

  • Print the input DataFrame.

  • Create a list of values for selection of rows.

  • Print the selected rows with the given values.

  • Next, print the rows that were not selected.

Example

 Live Demo

import pandas as pd df = pd.DataFrame(    {       "x": [5, 2, 1, 9],       "y": [4, 1, 5, 10],       "z": [4, 1, 5, 0]    } ) print "Input DataFrame:
", df values = [1, 2] print "Selected Rows:
", df[df['x'].isin(values)] print "Uselected Rows:
", df[~df['x'].isin(values)]

Output

Input DataFrame:    x  y  z 0  5  4  4 1  2  1  1 2  1  5  5 3  9 10  0 Selected Rows:    x  y  z 1  2  1  1 2  1  5  5 Unselected Rows:    x  y  z 0  5  4  4 3  9 10  0
Updated on: 2021-08-30T09:36:20+05:30

2K+ Views

Kickstart Your Career

Get certified by completing the course

Get Started
Advertisements