如何截断2D numpy数组的值

Question

我有一个二维的numpy数组（uint16），如何将所有大于某个值（比如255）的元素截断为该值？其他值必须保持不变。使用嵌套循环似乎是低效和笨拙的。

- nobody

Answer 1

其实有一个特定的方法可以实现这个功能，那就是'clip':

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

输出：

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

- Andrea Zonca

Answer 2

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

输出结果将会是：

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

- JBernardo

Answer 3

如果你的问题与JBernardo的答案不太相关，更一般的方法是这样做： （经过编辑，我的答案现在基本上和他的一样）

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

这里有一个很好的numpy布尔索引介绍链接。

- ImAlsoGreg