how to filter dataframe by column values by items in set code example
Example 1: filter a column in pandas by list of values
In [1]: df = pd.DataFrame({'A': [5,6,3,4], 'B': [1,2,3,5]})
In [2]: df
Out[2]:
A B
0 5 1
1 6 2
2 3 3
3 4 5
In [3]: df[df['A'].isin([3, 6])]
Out[3]:
A B
1 6 2
2 3 3
Example 2: filter df by column value
# does year equals to 2002?
# is_2002 is a boolean variable with True or False in it
>is_2002 = gapminder['year']==2002
>print(is_2002.head())
0 False
1 False
2 False
3 False
4 False