要将 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 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录