如何删除窗口内NaN值周围的数组中的点

import numpy as np

def remove_nan_points(array, window_size):
    height, width = array.shape
    half_window = window_size // 2
    
    for i in range(half_window, height - half_window):
        for j in range(half_window, width - half_window):
            if np.isnan(array[i, j]):
                continue
            
            window = array[i - half_window:i + half_window + 1, j - half_window:j + half_window + 1]
            if np.isnan(window).any():
                array[i, j] = np.nan  # 设置为NaN值
                # 或者直接删除该点
                # array = np.delete(array, (i, j), axis=(0, 1))
    
    return array