numpy change array values when mask is one

Question

i'm new to numpy and i'm running into trouble.

I've got two numpy arrays, img and thr:

>>>img.shape
(2448, 3264, 3)
>>>thr.shape
(2448, 3264)

And i want to do something like this: set img[x,y] = [255,255,255] only when thr[x,y] is not 0

I tried iterating over the array and do it myself but it takes a long time, so i really need the C underneath numpy. I also took a look to masked arrays but i didn't understand how to use them.

Thanks!

unutbu · Accepted Answer · 2012-12-01 12:29:08Z

4

Using NumPy assignment to an indexed array:

img[thr != 0] = [255,255,255]

edited Dec 1, 2012 at 12:29

answered Dec 1, 2012 at 12:23

unutbu

886k197 gold badges1.9k silver badges1.7k bronze badges

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

Collectives™ on Stack Overflow

numpy change array values when mask is one

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related