numpy.clip使数组中的值保持在一定区间内

numpy.clip使数组中的值保持在一定区间内

np.clip()

给定一个区间范围,区间范围外的值将被截断到区间的边界上。例如,如果指定的区间是 [-1,1],小于-1 的值将变为-1,而大于 1 的值将变为 1。

array = np.array([10, 7, 4, 3, 2, 2, 5, 9, 0, 4, 6, 0])
print(np.clip(array,2,6))
#输出:[6 6 4 3 2 2 5 6 2 4 6 2]

小于2的元素变为2,大于6的元素变为6,一行代码的简洁和高效远超这种写法:

array[array<2]=2
array[array>6]=6
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章