要从第一个元素为NaN的NumPy数组中删除行,可以使用以下步骤:
np.isnan()
函数检查数组中的NaN值,并返回一个布尔数组:nan_rows = np.isnan(arr[:, 0])完整的代码如下所示:
import numpy as np
arr = np.array([[np.nan, 1, 2],
[3, 4, 5],
[np.nan, 6, 7],
[8, 9, 10]])
nan_rows = np.isnan(arr[:, 0])
filtered_arr = arr[~nan_rows]
print(filtered_arr)
输出结果为:
[[ 3. 4. 5.]
[ 8. 9. 10.]]
在这个例子中,我们创建了一个包含NaN值的NumPy数组。然后,我们使用np.isnan()
函数检查数组的第一列是否包含NaN值,并将结果存储在布尔数组nan_rows
中。最后,我们使用布尔数组nan_rows
来选择需要保留的行,并将结果存储在filtered_arr
中。输出结果为删除了第一个元素为NaN的行的新数组。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云