Issue while using query in python

Question

Here is my data

cars = {'Brand': ['Honda Civic','Toyota Corolla','Ford Focus','Audi A4'],
        'Price': [22000,25000,27000,35000]
        }

df = pd.DataFrame(cars, columns = ['Brand', 'Price'])

I would like to input data to be filtered using i via pandas query.

While this works-

    i = 'Honda Civic'
   df[df['Brand'] == i]

While, if I try do it via query as shown below, it doesn't work -

i = 'Honda Civic'
(df
.query('Brand' == i))

How should I modify the query command to make it work?

my bad: df.query("Brand==@i") remove the quotes around Brand — sammywemmy
– sammywemmy, Commented Aug 24, 2020 at 21:12
This is my code i = ['Honda Civic'] (df. query(Brand==@i)) . I still get invalid syntax error :( — Manasi
– Manasi, Commented Aug 24, 2020 at 21:14
I just made an edit. Have a look at previous comment. df.query("Brand==@i"). and no need to wrap it in a list, if it is just one item. — sammywemmy
– sammywemmy, Commented Aug 24, 2020 at 21:14

aggis · Accepted Answer · 2020-08-24 21:16:32Z

3

I always prefer:

df[df['Brand'] == i]

This is much easier for me, and I try to avoid the "query" method of Pandas DataFrames. Especially since you can expand on the resulting DataFrame slice. For example, if you had other columns such as "State" and "Price", you could groupby + aggregate like so:

df[df['Brand'] == i].groupby('State')['Price'].sum()

answered Aug 24, 2020 at 21:16

aggis

6984 silver badges9 bronze badges

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

Collectives™ on Stack Overflow

Issue while using query in python

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related