在 MATLAB 中替换矩阵值

Question

我有一个矩阵比如

M = [ 1 3 2 4;
      3 3 2 1;
      2 4 1 3]

有一个基础A = [ 1 2 3 4];

我还有另一个基地B = [103 104 105 106];

我需要用 M 中的 B 值替换 A 的值。所以我的新 M 应该是：

M1 = [ 103  105 104 106;
       105  105 104 103;
       104  106 103 105];

元素是随机数，所以我需要在 A 和 B 之间使用索引一对一的连接。 我应该提一下吗？当然没有循环：D 谢谢

Answer 1

如果您的基础是旧基础的简单功能，则它可能是微不足道的：

M1 = M + 102;

否则这是一种方式：

M1 = M
for t = 1:length(A)
    M1(M==A(t)) = B(t)
end

根据@Rody 的回答，另一种解决方案：

[idx,b] = ismember(M,A);      
M(idx) = B(b(idx))

不同的是，如果A 不包含M 的所有元素，这不会中断。 （如果是适当的依据，应该不会发生）。

Answer 2

这是一个适合你的单行：

sum(bsxfun(@times, bsxfun(@eq, M, reshape(A,1,1,[])), reshape(B,1,1,[])), 3)

相当快。

基准测试

这是基准测试代码：

%// bsxfun party
tic
for k = 1:10000
    M1 = sum(bsxfun(@times,bsxfun(@eq,M,reshape(A,1,1,[])),reshape(B,1,1,[])),3);
end
toc

%// Using ismember
tic
for k = 1:10000
    [idx,b] = ismember(M,A);      
    M(idx) = B(b(idx));
end
toc

%// Using a simple loop
tic
for k = 1:10000
    M1 = M;
    for t = 1:length(A)
        M1(M == A(t)) = B(t);
    end
end
toc

结果是：

Elapsed time is 0.030135 seconds.
Elapsed time is 0.094354 seconds.
Elapsed time is 0.007410 seconds.

所以这个单线比ismember 的优雅解决方案更快，但简单的（JIT 加速）循环胜过两者。令人惊讶，不是吗？ :)

Answer 3

如果您确定新的M 包含来自新基的仅 元素（旧的M 和旧的基相同），您可以使用ismember 的第二个输出：

 >> [~,b] = ismember(M,A);      
 >> M = B(b)
 M =
    103   105   104   106
    105   105   104   103
    104   106   103   105