Keyboard shortcuts

Press or to navigate between chapters

Press S or / to search in the book

Press ? to show this help

Press Esc to hide this help

DataFrame filter rows by value in list

import sys
import pandas as pd

filename = "planets.csv"
if len(sys.argv) == 2:
    filename = sys.argv[1]

df = pd.read_csv(filename)


planet_selector = df['Planet name'].isin(['Earth', 'Mars'])
print(planet_selector)
print()

planets = df[ planet_selector ]
print(planets)

Output:

0     False
1     False
2      True
3      True
4     False
5     False
6     False
7     False
8     False
9     False
10    False
Name: Planet name, dtype: bool

  Planet name  Distance (AU)   Mass
2       Earth            1.0  1.000
3        Mars            1.5  0.107