np.clip的用法

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

其中a是一个数组,后面两个参数分别表示最小和最大值,将数组中的元素限制在a_min, a_max之间,大于a_max的就使得它等于 a_max,小于a_min,的就使得它等于a_min。

a = np.arange(10)
np.clip(a, 1, 8)
Out[36]: array([1, 1, 2, 3, 4, 5, 6, 7, 8, 8])
a
Out[37]: array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
当out默认为None时,a不变,np.clip函数返回的数组没有赋给a。

np.clip(a, 3, 6, out=a)
Out[38]: array([3, 3, 3, 3, 4, 5, 6, 6, 6, 6])
a
Out[39]: array([3, 3, 3, 3, 4, 5, 6, 6, 6, 6])
当out指定为a时,会把np.clip函数的返回值赋给a。


猜你喜欢

转载自blog.csdn.net/zhuzuwei/article/details/80075501