NumPy 数组区间值替换技巧:将特定范围内的值设为同一值
要将 NumPy 数组中的值在某个区间设为同一个值,可以使用 NumPy 的条件表达式来实现。
下面是一个示例代码,假设我们有一个 NumPy 数组 arr,将所有小于等于 4 的值设为 -1:
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
arr[arr <= 4] = -1
print(arr)
输出结果为:
[-1 -1 -1 -1 5 6 7 8 9]
在这个示例中,我们使用了条件表达式 'arr <= 4' 来选取所有小于等于 4 的元素,并将其赋值为 -1。
原文地址: https://www.cveoy.top/t/topic/pigL 著作权归作者所有。请勿转载和采集!