Compare 2 columns in Pandas DataFrame with .loc [duplicate]

Question

I want to compare two columns with value (1) and list rows that satisfy this condition. Here is my code:

import pandas as pd

df = pd.DataFrame({'col':[0,1,1,0,1],
                   'col2':[0,1,0,1,0],
                   'ord':[0,1,2,3,4]
})

df1 = df.loc[df['col'] == 1 & df['col2'] == 1]

print(df1)

Expected output:

   col  col2  ord
0    1     1    1

jezrael · Accepted Answer · 2019-04-14 13:41:38Z

4

Add parentheses because priority precedence of & operator:

df1 = df.loc[(df['col'] == 1) & (df['col2'] == 1)]

edited Apr 14, 2019 at 13:41

answered Apr 14, 2019 at 13:40

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

Collectives™ on Stack Overflow

Compare 2 columns in Pandas DataFrame with .loc [duplicate]

1 Answer 1

Comments

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Linked

Related