Python - drop elements from DataFrame

Question

I have DataFrame where are few columns. I have also list of unique elements (elements from one of the columns).

I would like to leave only rows with elements in DataFrame which are included in my list. I tried to do that but all my ideas failed.

Below quick example:

list = ["a", "b"]

Col1 Col2 Col3
1    a     ok
2    b     nok
3    c     ok
4    d     ok
5    a     nok

So I want to keep only rows where are A and B (elements from my list).

Sunderam Dubey · Accepted Answer · 2022-07-13 14:00:38Z

4

Try this:

df.loc[df[column_you_want].isin(your_list)]

edited Jul 13, 2022 at 14:00

Sunderam Dubey

8,83312 gold badges26 silver badges43 bronze badges

answered Jul 13, 2022 at 12:26

bpfrd

1,0255 silver badges11 bronze badges

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

Comments

theophile · Accepted Answer · 2022-07-13 13:14:36Z

3

df = df.loc[df["Col2"].isin(["a", "b"])]

will give:

Col1 Col2 Col3
1    a     ok
2    b     nok
5    a     nok

edited Jul 13, 2022 at 13:14

answered Jul 13, 2022 at 12:27

theophile

2961 silver badge6 bronze badges

1 Comment

tcdejong Over a year ago

The resulting df will be a view, so mutating the df can give trouble down the line. You should use df.loc instead, so that the return type is a real DataFrame rather than a view.

Collectives™ on Stack Overflow

Python - drop elements from DataFrame

2 Answers 2

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related