numpy中np.clip函數(shù)的用法是什么

小億
126
2024-02-04 12:28:46

np.clip函數(shù)用于限制數(shù)組的取值范圍。它接受三個(gè)參數(shù):數(shù)組a、最小值min或最小值數(shù)組amin、最大值max或最大值數(shù)組amax。

當(dāng)傳入最小值和最大值時(shí),np.clip函數(shù)會(huì)將數(shù)組a中小于最小值的元素替換為最小值,大于最大值的元素替換為最大值,其他元素保持不變。

當(dāng)傳入最小值數(shù)組和最大值數(shù)組時(shí),np.clip函數(shù)會(huì)將數(shù)組a中對(duì)應(yīng)位置的元素限制在最小值和最大值數(shù)組的范圍內(nèi)。如果最小值數(shù)組中的元素小于最大值數(shù)組中的元素,則對(duì)應(yīng)位置的元素會(huì)被限制在最小值和最大值數(shù)組中的范圍內(nèi),否則不做任何修改。

示例如下:

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]

在第一個(gè)示例中,使用np.clip函數(shù)將數(shù)組a中小于2的元素替換為2,大于4的元素替換為4,結(jié)果為[2, 2, 3, 4, 4]。

在第二個(gè)示例中,使用np.clip函數(shù)將數(shù)組c中小于2的元素替換為2,大于4的元素替換為4,結(jié)果為[2, 4, 3, 4, 4]。

0