Examples of how to select a dataframe rows using a condition with pandas in python:
1 -- Create a simple dataframe
Lets start by creating a simple dataframe with pandas:
>>> import pandas as pd>>> data = {'Name':['Ben','Anna','Zow','Tom','John','Steve'], 'Age':[20,27,43,30,12,21], 'Sex':[1,0,0,1,1,1]}>>> df = pd.DataFrame(data)
returns
>>> dfAge Name Sex0 20 Ben 11 27 Anna 02 43 Zow 03 30 Tom 14 12 John 15 21 Steve 1
2 -- Select dataframe rows using a condition
Example lets select the rows where the column named 'sex' is equal to 1:
>>> df[ df['Sex'] == 1 ]Age Name Sex0 20 Ben 13 30 Tom 14 12 John 15 21 Steve 1
3 -- Select dataframe rows using two conditions
Another example using two conditions with & (and):
>>> df[ (df['Sex'] == 1) & (df['Age'] < 25 )]Age Name Sex0 20 Ben 14 12 John 15 21 Steve 1
