np.clip函數用于限制數組的取值范圍。它接受三個參數:數組a、最小值min或最小值數組amin、最大值max或最大值數組amax。
當傳入最小值和最大值時,np.clip函數會將數組a中小于最小值的元素替換為最小值,大于最大值的元素替換為最大值,其他元素保持不變。
當傳入最小值數組和最大值數組時,np.clip函數會將數組a中對應位置的元素限制在最小值和最大值數組的范圍內。如果最小值數組中的元素小于最大值數組中的元素,則對應位置的元素會被限制在最小值和最大值數組中的范圍內,否則不做任何修改。
示例如下:
import numpy as np
a = np.array([1, 2, 3, 4, 5])
b = np.clip(a, 2, 4)
print(b) # 輸出: [2 2 3 4 4]
c = np.array([1, 6, 3, 8, 5])
d = np.clip(c, [2, 2, 2, 2, 2], [4, 4, 4, 4, 4])
print(d) # 輸出: [2 4 3 4 4]
在第一個示例中,使用np.clip函數將數組a中小于2的元素替換為2,大于4的元素替換為4,結果為[2, 2, 3, 4, 4]。
在第二個示例中,使用np.clip函數將數組c中小于2的元素替換為2,大于4的元素替換為4,結果為[2, 4, 3, 4, 4]。