Replace string values in array by multiple conditions in Python

Question

I have an array: array = ['wood', 'glass', 'metal', 'glass', 'glass', 'wood', 'metal', 'wood', 'glass', 'glass']

and I want to replace each string by a condition, for example: each 'wood' replace by 'blue', 'glass' by 'red' and 'metal' by 'green'.

so I get: ['blue', 'red', 'green', 'red', 'red', 'blue', 'green', 'blue', 'red', 'red']

I'm trying to do something like: ['red' if el == 'glass' for el in array]

I don't know how to do multiple conditions or even if this method is the right thing to do?

please help :)

abc · Accepted Answer · 2021-01-05 20:43:36Z

2

You can use a dictionary to map old value to new values.

Example:

>>> name_mapping = {'wood':'blue'}
>>> res = [name_mapping.get(el, el) for el in array]
>>> res
['blue', 'glass', 'metal', 'glass', 'glass', 'blue', 'metal', 'blue', 'glass', 'glass']

dict.get with keep the current element in the result in case the element is not a key of the mapping dictionary.

answered Jan 5, 2021 at 20:43

abc

12k2 gold badges30 silver badges55 bronze badges

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

Comments

abe · Accepted Answer · 2021-01-05 20:44:27Z

1

Maybe create a dictionary:

dct = {'wood': 'blue', 'glass': 'red', 'metal': 'green'}
new_array = [dct[item] for item in array]

answered Jan 5, 2021 at 20:44

abe

9896 silver badges10 bronze badges

2 Comments

cicolus Over a year ago

This won't work when item is not one of the keys.

abe Over a year ago

@cicolus yes obviously. this was just a quick demonstration how to use dictionaries with a list comprehension

Collectives™ on Stack Overflow

Replace string values in array by multiple conditions in Python

2 Answers 2

Comments

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related