Numpy array - Replace array

Question

I have a numpy array a of shape [300, 3, 3], I like to change this array - [0.7, 0.3, 0.1] to [1, 1, 1], if occured in a

For example,

Input -

b = np.array([[[0.7, 0.3, 0.1], [0.6, 0.4, 0.2], [0.1, 0.2, 0.1]],
             [[0.7, 0.3, 0.1], [0.6, 1.2, 2.1], [1.1, 2.1, 1.1]]
             ])

Output-

np.array([[[1, 1, 1], [0.6, 0.4, 0.2], [0.1, 0.2, 0.1]],
         [[1, 1, 1], [0.6, 1.2, 2.1], [1.1, 2.1, 1.1]]
         ])

How to achieve this in easiest way, without use of looping, thanks

Aldrin Saurov Sarker · Accepted Answer · 2021-06-18 14:00:01Z

3

Use where method to replace elements of array.

b = np.where(b == [0.7, 0.3, 0.1], [1, 1, 1], b)

answered Jun 18, 2021 at 14:00

3333 silver badges15 bronze badges

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

Comments

dimay · Accepted Answer · 2021-06-18 13:56:19Z

1

Try this:

np.where(b == [0.7, 0.3, 0.1], [1, 1, 1], b)

answered Jun 18, 2021 at 13:56

dimay

2,8341 gold badge17 silver badges25 bronze badges