我有一个二维的numpy数组(uint16),如何将所有大于某个值(比如255)的元素截断为该值?其他值必须保持不变。使用嵌套循环似乎是低效和笨拙的。
其实有一个特定的方法可以实现这个功能,那就是'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)
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)
def trunc_to(my_array, limit): too_high = my_array > limit my_array[too_high] = limit这里有一个很好的numpy布尔索引介绍链接。