numpy_clip函数

  函数原型如下：

numpy.clip(a, a_min, a_max, out=None, **kwargs)

clip将数组a中的元素大小限制在[a_min, a_max]。

import numpy as np

a = np.arange(10)
b = np.clip(a, 1, 8)
print("a =", a)
print("b =", b)

执行结果：

a = [0 1 2 3 4 5 6 7 8 9]
b = [1 1 2 3 4 5 6 7 8 8]