Pandas Query Data With Categorical Variables

Pandas Query is a great way to filter your data. But what if you have categorical variables. In this case, you’re gonna want to treat them a bit differently.

The important part to remember is that you will need to put your python expression inside of a string. After that it’s all peaches and cream.

1. pd.DataFrame.query("your_column == 'your value')

Pseudo code: For a given column or columns return the values that resolve to true

Pandas Query With Categorical Variables

Let’s take a look at a code sample

Link to code

Official Documentation