使用numpy.clip(…)根据指定的min和max值将数据限定在一定范围内截断,函数原型如下:
numpy.clip(a, a_min, a_max, out=None, **kwargs)
【sample】
In [1]: import numpy as np
In [2]: dat = np.array([i for i in range(-5, 6)])
In [3]: dat
Out[3]: array([-5, -4, -3, -2, -1, 0, 1, 2, 3, 4, 5])
# 指定min=-3, max=2,将数据限制在-3~2(包括-3和2)
In [4]: dat = np.clip(dat, -3, 2)
In [5]: dat
Out[5]: array([-3, -3, -3, -2, -1, 0, 1, 2, 2, 2, 2])