强制Numpy数组中的元素在指定范围内

时间:2011-08-12 03:02:28

标签: python numpy

我有一个numpy数组,我想强制每个小于零的元素为零,并且255以上的每个元素都将被强制为255.

例如。 x =(-1,7,255,299)=> (0,7,255,255)

是否有一个不太复杂的单行程可以实现这个目标?

1 个答案:

答案 0 :(得分:8)

答案是numpy.clip

numpy.clip(x, 0, 255)

关于标题中张贴的问题:不要。您可以使用vectorize将lambda函数应用于每个元素,但这很少是最佳选择。