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
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章