numpy 二维数组把小于0.15的值负值为正无穷
时间: 2024-04-12 14:32:32 浏览: 188
您可以使用NumPy的条件判断和赋值操作来实现将小于0.15的值设为正无穷。以下是一个示例代码:
```python
import numpy as np
# 创建一个示例的二维数组
arr = np.array([[0.1, 0.2, 0.3],
[0.4, 0.5, 0.6],
[0.7, 0.8, 0.9]])
# 将小于0.15的值设为正无穷
arr[arr < 0.15] = np.inf
print(arr)
```
输出结果为:
```
[[ inf 0.2 0.3]
[ 0.4 0.5 0.6]
[ 0.7 0.8 0.9]]
```
在这个示例中,我们首先创建了一个示例的二维数组 `arr`。然后,使用条件判断 `arr < 0.15` 来获取小于0.15的元素的布尔掩码,然后将这些元素赋值为正无穷 `np.inf`。最后,打印输出数组 `arr` 的值。如此一来,小于0.15的值就被替换为了正无穷。
阅读全文