pandas dataframe based on column value code example

Example 1: panda - subset based on column value

import pandas as pd
import numpy as np
df = pd.DataFrame({'A': 'foo bar foo bar foo bar foo foo'.split(),
                   'B': 'one one two three two two one three'.split(),
                   'C': np.arange(8), 'D': np.arange(8) * 2})
print(df)
sub_df = df.loc[df['A'] == 'foo'] # Subset based on specific A value

Example 2: selecting a specific value and corrersponding value in df python

#To select rows whose column value equals a scalar, some_value, use ==:df.loc[df['favorite_color'] == 'yellow']

Tags:

Misc Example