Error when selecting rows in pandas dataframe based on column value

Question

I have a dataframe df which looks like this:

col1  col2 col3
 A     45    4
 A     3     5
 B     2     5

I want to make a separate dataframe, df2 , which only has the rows where col in df equals A . Hence it should look like:

col1  col2 col3
 A     45    4
 A     3     5

I just use df2=df1.loc[df1['col1']=='A'] . However this returns the error: ValueError: Cannot index with multidimensional key . Any idea what is going wrong here?

Answer 1

What you tried works for me, you can try this:

df2 = df[df.col1 == 'A']

Output

  col1  col2    col3
0   A   45      4
1   A   3       5

Edit Tested on pandas version

pd.__version__
'1.2.4'

Answer 2

Just try:

df.query("col1=='A'")