在 Matlab 中用两个值替换索引

Question

我正在尝试在 Matlab 中解决以下问题：

函数replace_me 定义如下：function w = replace_me(v,a,b,c)。第一个输入参数v 是一个向量，而a、b 和c 都是标量。该函数将v 中等于a 的每个元素替换为b 和c。

例如命令

x = replace_me([1 2 3],2,4,5);

使x 等于[1 4 5 3]。如果省略了c，它将用b 的两个副本替换出现的a。如果b 也被省略，它会将每个a 替换为两个零。”

这是我目前所拥有的：

function [result] = replace_me(v,a,b,c)
    result = v;
    for ii = 1:length(v)
        if result(ii) == a
            result(ii) = b;
            result(ii +1) = c;
        end
    end
end

该函数将v 中与a 相同的值替换为b，但我无法将a 替换为b 和c。任何帮助将非常感激。

编辑：

我已经更新了我的代码：

function [v] = replace_me(v,a,b,c)
for ii = 1:length(v)
  if v(ii) == a
    v = horzcat(v(1:ii-1), [b c], v(ii+1:end));
    fprintf('%d',v);
  elseif v(ii) == a && length(varargin) == 3
    v = horzcat(v(1:ii-1), [b b], v(ii+1:end));
    fprintf('%d',v);
  elseif v(ii) == a && length(varargin) == 2
    v = horzcat(v(1:ii-1), [0 0], v(ii+1:end));
    fprintf('%d',v);
  else

      fprintf('%d',v);
  end

end
end

我在尝试 replace_me([1 2 3 4 5 6 7 8 9 10], 2, 2) 时收到以下错误：

replace_me([1 2 3 4 5 6 7 8 9 10],2,2) 12345678910 使用 replace_me 时出错（第 4 行） 输入参数不足。

Answer 1

这是另一种选择：（最初由@A. Donda 提供answered）

这样做的好处是替换多次出现

function [result] = replace_me(v,a,b,c)
    if nargin == 3
        c = b;
    end
    if nargin == 2
        c = 0;
        b = 0;
    end
    result = v;
    equ = result == a;
    result = num2cell(result);
    result(equ) = {[b c]};
    result = [result{:}];
end

结果：

>> x = replace_me([1 2 3],2,4,5)

x =

 1     4     5     3

>> x = replace_me([1 2 3],2)

x =

 1     0     0     3

>> x = replace_me([1 2 3],2,4)

x =

 1     4     4     3

>> x = replace_me([1 2 3 2], 2, 4, 5)

x =

 1     4     5     3     4     5

Answer 2

通过计算a 在v 中出现的次数，您可以计算得到的向量length(v) + sum(v == a) 的大小，然后您可以遍历向量并在每次找到匹配项时插入[b c]。您应该在每次插入新元素时将计数器 ii 增加 2，否则如果 a=b 您将使用 a 填充向量的剩余部分。这不是一个有效的算法，但它类似于你的实现

result = v;
rep = sum(result == a);

if rep == 0
  return
end

for ii = 1:(length(result) + rep)
  if result(ii) == a
    result = [result(1:ii-1), [b c], result(ii+1:end)];
    ii = ii + 2;
  end
end

一种有效的替代方法（无需调整矢量大小）可能是：

rep = sum(v == a);

if rep == 0
    result = v;
    return;
end

result = zeros(1, length(v) + rep);

idx = 1;
for ii = 1:length(v)
    if v(ii) == a
       result(idx) = b;
       result(idx+1) = c;
       idx = idx + 2;
    else 
       result(idx) = a;
       idx = idx + 1;
    end
end

Answer 3

可以是这样的：

function result = replace_me(v, a, b, c)
  result = [];
  for ii = 1:length(v)
    if v(ii) == a
      result = [result, b, c];
    else
      result = [result, v(ii)];
    end
  end
end

关于错误“Error using replace_me (line 4) Not enough input arguments.”，这仅仅是因为你用4个输入参数编写了函数，而你只用3个输入参数调用。