把numpy中的数组元素设置为nan
时间: 2024-05-13 18:19:14 浏览: 70
Python快速转换numpy数组中Nan和Inf的方法实例说明
5星 · 资源好评率100%
在 NumPy 中,可以使用 `numpy.nan` 函数将数组中的元素设置为 NaN(Not a Number),代码如下:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
arr[np.random.choice(arr.size, replace=False, size=2)] = np.nan
print(arr)
```
这里的 `np.random.choice` 函数用于随机选择数组中的两个元素,然后将它们设置为 NaN。输出结果如下:
```
[ 1. 2. 3. nan nan]
```
如果想要将整个数组的所有元素都设置为 NaN,可以使用 `numpy.full` 函数创建一个与原数组形状相同的数组,并将其元素全部设置为 NaN,代码如下:
```python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
nan_arr = np.full(arr.shape, np.nan)
print(nan_arr)
```
输出结果如下:
```
[nan nan nan nan nan]
```
阅读全文