如果你想要一个新的数组
result
,其中包含了
arr
的副本,只有当
arr < 255
时,否则为
255
。
result = np.minimum(arr, 255)
更一般地说,对于一个下界和/或上界:
result = np.clip(arr, 0, 255)
如果你只想访问超过255的值,或者需要更复杂的操作,@mtitan8的回答更通用,但是np.clip和np.minimum(或np.maximum)对于你的情况来说更好且速度更快。
In [292]: timeit np.minimum(a, 255)
100000 loops, best of 3: 19.6 µs per loop
In [293]: %%timeit
.....: c = np.copy(a)
.....: c[a>255] = 255
.....:
10000 loops, best of 3: 86.6 µs per loop
如果你想在原地进行操作(即修改`arr`而不是创建`result`),你可以使用`np.minimum`的`out`参数。
np.minimum(arr, 255, out=arr)
或者
np.clip(arr, 0, 255, arr)
(
out=
名称是可选的,因为参数的顺序与函数的定义相同。)
对于原地修改,布尔索引速度提高了很多(无需单独创建和修改副本),但仍然不如
minimum
快。
In [328]: %%timeit
.....: a = np.random.randint(0, 300, (100,100))
.....: np.minimum(a, 255, a)
.....:
100000 loops, best of 3: 303 µs per loop
In [329]: %%timeit
.....: a = np.random.randint(0, 300, (100,100))
.....: a[a>255] = 255
.....:
100000 loops, best of 3: 356 µs per loop
假设你想要限制数值的最小和最大值,如果没有使用 clip
函数,你需要重复两次操作,类似于下面的代码:
np.minimum(a, 255, a)
np.maximum(a, 0, a)
或者,
a[a>255] = 255
a[a<0] = 0