Extract rows based on a condition - Pandas

Question

I want to extract all rows from a df where the 'route_id' = 6A.

This is what I have tried but it returns a syntax error:

 df.loc(df['route_id'] = 6A)

Any tips? Thanks in advance.

Answer 1

Looks like you've used the assignment = instead of equality test == . The .loc needs square brackets and I also added some quotes around the '6A'

df.loc[df['route_id'] == '6A']