np.clip()
时间: 2023-07-04 12:07:10 浏览: 125
PSO.zip
np.clip() 是 NumPy 库中的一个函数,用于将数组中的元素限制在一个指定范围内。它的语法如下:
```
np.clip(a, a_min, a_max, out=None)
```
其中,参数 a 表示需要限制范围的数组,a_min 和 a_max 分别表示限制范围的下界和上界,如果 a[i] < a_min,则 a[i] = a_min;如果 a[i] > a_max,则 a[i] = a_max;否则 a[i] 不变。如果指定了参数 out,则将结果保存到 out 中,否则返回一个新的数组。
例如,对于一个数组 a = [1, 2, 3, 4, 5],我们可以将它限制在 [2, 4] 的范围内,代码如下:
```
import numpy as np
a = np.array([1, 2, 3, 4, 5])
a_clip = np.clip(a, 2, 4)
print(a_clip) # 输出 [2 2 3 4 4]
```
这样,原数组 a 中小于 2 的元素被限制为 2,大于 4 的元素被限制为 4,而 2 到 4 之间的元素不变。
阅读全文