Find if column contains a certain value in panda

CoderGirl

Edited On 29 AUGUST 2022

Category: Python

1 Answers
170 Views
0 Likes

Example I have a dataframe with the following columns:

import pandas as pd
df = pd.DataFrame({
    'A': [1, 4, 7, 1, 4],
    'B': [2, 5, 8, 2, 5],
    'C': [3, 6, 9, 3, 6]
})

Data when represented in Table format is:

Now I want to find the index of the column that contains the value 5.

How to find the index of column?

1 Replies

Answer Now

You can use where function of numpy to find index of the column that contains particular value. Syntax for that is:

numpy.where(condition[, x, y])

Now for the example you gave above, we can use below code to find index of column in panda:

import numpy as np
col_index = pd.DataFrame(np.where(df.eq(5))[1] + 1, columns=['col_index'])

Output:

   col_index
0          2
1          2

Published on 29 AUGUST 2022 at 9:1

0 Likes Reply