matlab - 如何从 Matlab 中的向量中删除可变数量的连续 NaN 值？

Question

我有一个值向量，如下所示：

1
2
3
NaN
4
7
NaN
NaN
54
5
2
7
2
NaN
NaN
NaN
5
54
3
2
NaN
NaN
NaN
NaN
4
NaN

我该如何使用

interp1

以这种方式只插入可变数量的连续 NaN 值？例如，我只想插入那些最多有三个连续 NaN 值的 NaN 值。所以NaN，NaN NaN和NaN NaN NaN将被插值，但不是NaN NaN NaN NaN。

谢谢你的帮助=）

PS如果我不能用interp1做到这一点，有什么想法可以用另一种方式做到这一点吗？=)

举个例子，我给出的向量将变为：

1
2
3
interpolated
4
7
interpolated
interpolated
54
5
2
7
2
interpolated
interpolated
interpolated
5
54
3
2
NaN
NaN
NaN
NaN
4
interpolated

Answer 1

首先，找出所有NaN值序列的位置和长度：

nan_idx = isnan(x(:))';
nan_start = strfind([0, nan_idx], [0 1]);
nan_len = strfind([nan_idx, 0], [1 0]) - nan_start + 1;

接下来，找到NaN不插值的元素的索引：

thr = 3;
nan_start = nan_start(nan_len > thr);
nan_end = nan_start + nan_len(nan_len > thr) - 1;
idx = cell2mat(arrayfun(@colon, nan_start, nan_end, 'UniformOutput', false));

现在，插入所有内容并将不应该插入的元素替换为NaN值：

x_new = interp1(find(~nan_idx), x(~nan_idx), 1:numel(x));
x_new(idx) = NaN;

Answer 2

我知道这在 matlab 中是一个坏习惯，但我认为这种特殊情况需要一个循环：

function out = f(v)
  out = zeros(numel(v));
  k = 0;
  for i = 1:numel(v)
    if v(i) ~= NaN
      if k > 3
        out(i-k:i - 1) = ones(1, k) * NaN;             
      else
        out(i-k: i - 1) = interp1();%TODO: call interp1 with right params
      end
      out(i) = v(i)
      k = 0

    else
      k = k + 1 % number of consecutive NaN value encoutered so far
    end

 end