在numpy数组中内插NaN值

在numpy数组中内插NaN值,第1张

在numpy数组中内插NaN值

让我们首先定义一个简单的辅助函数,以使其更直接地处理NaN的索引和逻辑索引:

import numpy as npdef nan_helper(y):    """Helper to handle indices and logical indices of NaNs.    Input:        - y, 1d numpy array with possible NaNs    Output:        - nans, logical indices of NaNs        - index, a function, with signature indices= index(logical_indices),          to convert logical indices of NaNs to 'equivalent' indices    Example:        >>> # linear interpolation of NaNs        >>> nans, x= nan_helper(y)        >>> y[nans]= np.interp(x(nans), x(~nans), y[~nans])    """    return np.isnan(y), lambda z: z.nonzero()[0]

现在

nan_helper(.)
可以像这样使用:

>>> y= array([1, 1, 1, NaN, NaN, 2, 2, NaN, 0])>>>>>> nans, x= nan_helper(y)>>> y[nans]= np.interp(x(nans), x(~nans), y[~nans])>>>>>> print y.round(2)[ 1.    1.    1.    1.33  1.67  2.    2.    1.    0.  ]

-–
尽管指定一个单独的函数来执行以下 *** 作似乎有点过头了:

>>> nans, x= np.isnan(y), lambda z: z.nonzero()[0]

它最终将支付股息。

因此,每当您处理与NaNs相关的数据时,只需将其所需的所有(新的与NaN相关的新功能)封装在某些特定的辅助函数下即可。您的代码库将更易于理解,因为它遵循易于理解的习惯用法。

实际上,内插法是了解如何完成NaN处理的一个很好的上下文,但是在其他各种上下文中也使用了类似的技术。



欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5639473.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-16
下一篇 2022-12-16

发表评论

登录后才能评论

评论列表(0条)

保存