NumPy 获取 nan 的索引 - Python

NumPy是Python中重要的科学计算库之一，它提供了处理数组数据的高级功能。在数据处理中，有时候需要获取数组中为NaN（Not a Number）的元素的索引值。本文将介绍使用NumPy获取NaN索引的方法。

NumPy获取NaN索引的方法

使用NumPy中的isnan()函数可以检查数组中的每个元素是否为NaN，返回一个布尔类型的数组，其中含有True值的索引即为NaN的索引。下面是示例代码：

import numpy as np

arr = np.array([1, np.nan, 3, np.nan, 5, 6])
nan_index = np.isnan(arr)
print(nan_index)

输出结果：

[False  True False  True False False]

上面的代码中，我们首先定义了一个NumPy数组，其中包含两个NaN。然后使用isnan()函数检查每个元素是否为NaN，并将结果赋值给一个变量nan_index。输出结果可以看到，数组中第2和第4个元素的值为True，表明它们是NaN。

接下来，我们可以通过NumPy中的where()函数获取所有NaN的索引，代码如下：

nan_index = np.where(np.isnan(arr))
print(nan_index)

输出结果：

(array([1, 3], dtype=int64),)

由于where()函数返回的结果是一个包含元素索引的元组，因此需要在print语句中使用索引0来获取结果。输出结果可以看到，数组中第2个和第4个元素的索引分别为1和3。

总结

本文介绍了使用NumPy获取数组中NaN值的索引的方法，首先使用isnan()函数检查每个元素是否为NaN，然后使用where()函数获取所有NaN的索引。这些方法可以帮助我们在数据处理中更轻松地找到并处理NaN值。