change every value in a numpy array with a condition

Question

I have a 2d array that i got from and image, for now it has 0s and 255s, I want to change all the 255s into 1s, this is a very easy task for a for loop.

for i in range(lenX):
    for j in range(lenY):
        if img[i,j]==255:
            img[i,j] = 1

here img is my array. I am pretty sure that there is a simpler way to do it using some kind of numpy function or something. but I looked every where I couldn't find.

If you know how to do this easily.. please help me

You shouldn't use explicit for loops like that because the operation is vectorized. — gunesevitan
– gunesevitan, Commented Feb 12, 2019 at 7:56

Martin · Accepted Answer · 2019-02-12 07:55:56Z

6

This way you can modify matrix with conditions without loops

img[img==255]=1

answered Feb 12, 2019 at 7:55

Martin

3,3952 gold badges21 silver badges42 bronze badges

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

Comments

Sociopath · Accepted Answer · 2019-02-12 08:00:03Z

5

Use np.where

import numpy as np 

a = np.array([[1,9,1],[12,15,255],[255,1,245],[23,255,255]]) 
a = np.where(a==255, 1, a)
print(a)

Output:

[[  1   9   1]                                                                                                                                                    
 [ 12  15   1]                                                                                                                                                    
 [  1   1 245]                                                                                                                                                    
 [ 23   1   1]]

edited Feb 12, 2019 at 8:00

answered Feb 12, 2019 at 7:54

Sociopath

13.4k22 gold badges53 silver badges82 bronze badges

Collectives™ on Stack Overflow

change every value in a numpy array with a condition

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related