Converting string into condition for pandas dataframe

Question

I have a condition for a pandas dataframe written in a string. Something like this:

"(data['Variable1'] == 1) & (data['Variable2'] == 2)"

Is there a way to apply this condition without using eval() function?

Expected result:

data = data[(data['Variable1'] == 1) & (data['Variable2'] == 2)]

Out of curiousity, this looks like a perfect usecase for eval, what's the reason you're not using it? — Erfan
– Erfan, Commented Feb 26, 2020 at 12:20
eval works fine on a normal script but i have an error when using eval() within an imported module of my main script. And i honestly don't know why that happens.. — Ricardo Vilaça
– Ricardo Vilaça, Commented Feb 26, 2020 at 12:27

Mahendra Singh · Accepted Answer · 2020-02-26 12:11:00Z

6

You can use pandas query to filter required rows

your_query_string = "Variable1 == 1 & Variable2 == 2"
data = data.query(your_query_string)

answered Feb 26, 2020 at 12:11

Mahendra Singh

5282 silver badges9 bronze badges

Sign up to request clarification or add additional context in comments.

1 Answer 1