How to truncate values ​​of a 2D numpy array

Question

I have a two-dimensional numpy array (uint16), how can I truncate all values ​​above a certain barrier (say 255) to this barrier? The remaining values ​​should remain unchanged. Using a nested loop seems inefficient and awkward.

Answer 1

import numpy as np my_array = np.array([[100, 200], [300, 400]],np.uint16) my_array[my_array > 255] = 255 

the conclusion will be

 array([[100, 200], [255, 255]], dtype=uint16) 

Answer 2

actually there is a specific method for this, 'clip':

 import numpy as np my_array = np.array([[100, 200], [300, 400]],np.uint16) my_array.clip(0,255) # clip(min, max) 

exit:

 array([[100, 200], [255, 255]], dtype=uint16) 

Answer 3

In case your question was not related to bit depth, like JBernardo's answer, a more general way to do this would be something like this: (after editing, my answer is now almost the same as hiss)

  def trunc_to (my_array, limit):
     too_high = my_array> limit
     my_array [too_high] = limit

Here's a nice link to enter numup bool indexing.