Select rows from a pandas dataframe using a set of values

Question

I have a dataframe with a column named label_id which is a string value. I also have a set of label_id values in required_labels. I would like to select the rows of the dataframe where the label_id value is contained in the set.

I understand that I need to use df.loc for this, but when I try to generate a boolean mask that I can pass into df.loc, I get an error as below

boolean_mask = df['label_id'] in required_labels
TypeError: 'Series' objects are mutable, thus they cannot be hashed

What is the right way to do this?

imburningbabe · Accepted Answer · 2022-09-28 13:15:16Z

7

You have to use the series built-in func .isin() The correct syntax is:

boolean_mask = df['label_id'].isin(required_labels)

answered Sep 28, 2022 at 13:15

imburningbabe

8121 gold badge4 silver badges13 bronze badges

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

Collectives™ on Stack Overflow

Select rows from a pandas dataframe using a set of values

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related