Pandas Filtering row based on value using column index

Question

I am trying to filter a Dataframe based on value in a column

cust_id, prod_type
101, prod_A
102, prod_A
102, prod_B
103, prod_F
103, prod_A
104, prod_D

I am trying to filter based on column index as below:

df.loc[df.columns[1].eq('prod_A')]

It throws an error AttributeError: 'str' object has no attribute 'eq'

jezrael · Accepted Answer · 2020-04-28 06:24:30Z

3

You can select second column by positions by DataFrame.iloc:

df.loc[df.iloc[:, 1].eq('prod_A')]

Your solution should be changed with df[], because df.columns[1] return column name:

df.loc[df[df.columns[1]].eq('prod_A')]

answered Apr 28, 2020 at 6:24

jezrael

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

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

1 Answer 1