How can I find index of rows just same as a array from a pandas dataframe?

Question

I want to find indexes of some rows same as a array from pandas dataframe.

My dataframe is like below

df = pd.DataFrame({'a':[True, False, True, True],'b':[False, True, True, True],'c':[True, True, False, True],'d':[True, False, True, False]})

the input value that I'll enter is like row = [True, True, False, True], and I want get [2].

I tried some with for like this

index = []
for i in range df.shape[0]:
    if df.iloc[i,:].tolist() == row
        index.append(i)

It works but extremely slow for my work. Is there any method for this?

Chris Adams · Accepted Answer · 2019-06-03 11:07:25Z

4

Use DataFrame.eq and Series.all:

df[df.eq(row).all(1)]

Example

df = pd.DataFrame({'a':[True, False, True, True],'b':[False, True, True, True],'c':[True, True, False, True],'d':[True, False, True, False]})
row = [True, True, False, True]

df[df.eq(row).all(1)]

[out]

      a     b      c     d
2  True  True  False  True

If needed, chain on index.to_list() for list of indices:

df[df.eq(row).all(1)].index.to_list()

[2]

edited Jun 3, 2019 at 11:07

answered Jun 3, 2019 at 11:01

Chris Adams

18.7k4 gold badges26 silver badges44 bronze badges

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

1 Comment

Chris Adams Over a year ago

@Luterino no problem buddy, glad it helps

Collectives™ on Stack Overflow

How can I find index of rows just same as a array from a pandas dataframe?

1 Answer 1

Example

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Example

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related