numpy.clip使数组中的值保持在一定区间内

2022-09-02 20:08:21 浏览数 (1)

numpy.clip使数组中的值保持在一定区间内

np.clip()

给定一个区间范围,区间范围外的值将被截断到区间的边界上。例如,如果指定的区间是 [-1,1],小于-1 的值将变为-1,而大于 1 的值将变为 1。

代码语言:javascript复制
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,一行代码的简洁和高效远超这种写法:

代码语言:javascript复制
array[array<2]=2
array[array>6]=6

0 人点赞