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