Python - If column name contains a specific string then change the values in that column otherwise the values remain

Question

I'm trying to write some code which looks at the column names and sets the column value to 0 if that column name contains a specific string. I have a dataset with columns similar to this:

date apples_sold bananas_sold pineapples_sold orange_sold

I want to update any columns with the string "apple" to make every row 0 for that column. If the column doesn't contain "apple" then the value should remain.

I tried looping through the column names but am unsure the syntax as I don't want to hardcode the specific column names.

Quang Hoang · Accepted Answer · 2021-05-12 14:47:41Z

3

Try loc update with .str.contains:

df.loc[:, df.columns.str.contains('apple')] = 0

answered May 12, 2021 at 14:47

Quang Hoang

151k11 gold badges64 silver badges86 bronze badges

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

1 Comment

T D Over a year ago

That's amazing. I wish I asked this before spending hours on this. So eloquent, yet simple. Thank you!

Collectives™ on Stack Overflow

Python - If column name contains a specific string then change the values in that column otherwise the values remain

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related