如何在窗口中删除围绕 NaN 值的数组中的点

Question

我将 numpy 数组中的大纲替换为

signal[signal > high_limit] = np.nan

我可以定位那些 NaN 值的位置

nan_spots = np.where(np.isnan(signal))

但我坚持编写一个简单的代码来删除所有 NaN 值周围的相邻点，方法是用 NaNs 替换它们，在大纲的小窗口 +-1 左右。

Answer 1

import numpy as np
border = 1
signal = np.array([1,2,3,4,np.nan, np.nan, 6,8,0,np.nan])
has_nan_nb = lambda i: any(np.isnan([signal[k] for k in range(i-border, i+border+1) if k!=i and k>=0 and k<len(signal)]))
signal[np.vectorize(has_nan_nb)(np.arrange(len(signal)))] = np.nan

这在 1D 问题中很容易。