gabrielvie
4/3/2018 - 3:35 AM

Pandas - Filter DataFrames and Series

How do I filter pandas dataframes and series.

import pandas as pd

movies = pd.read_csv('http://bit.ly/imdbratings')

# AND condition
movies[(movies.duration >= 200) & (movies.genre == 'Drama')]

# OR condition
movies[(movies.duration >= 200) | (movies.genre == 'Drama')]

# IN to one or more than one values
movies[movies.genre.isin(['Crime', 'Drama', 'Action'])]
import pandas as pd

movies = pd.read_csv('http://bit.ly/imdbratings')


# FIRST WAY
booleans = []
for length in movies.duration:
    if length >= 200:
        booleans.append(True)
    else:
        booleans.append(False)
    # OR
    # > booleans.append(length >= 200)

is_long = pd.Series(booleans)

movies[is_long]

# SECOND WAY
movies[movies.duration >= 200]

# Filter using list in
movies[movies.genre.isin(['Crime', 'Action'])]

# Filter no null values
movies[movies.genre.notnull()]