Matlab 中的逐元素 if - 它们存在吗？

Question

假设我有以下基本 if 语句：

if (A ~= 0)
   % do something like divide your favorite number by A
else
   % do something like return NaN or infinity
end

问题是 A 不是一个简单的数字，而是一个向量。 如果 A 中没有元素为 0，Matlab 返回 true。我正在寻找的是矢量化的？ 为 A 中的每个元素执行上面的 if 语句的方法。

实际上，我只是想尽快执行此操作。

Answer 1

向量化 if 不存在，但有一些选项。 如果要测试所有或任何元素是否为 true，请使用 all 或 any 函数。

以下是有条件地修改矩阵值的一个示例：

b = A ~= 0;      % b is a boolean matrix pointing to nonzero indices
                 % (b could be derived from some other condition,
                 %  like b = sin(A)>0
A(b) = f(A(b))   % do something with the indices that pass
A(~b) = g(A(~b)) % do something else with the indices that fail

Answer 2

B = zeros(size(A));
B(A~=0) = FAV./A(A~=0);  
B(A==0) = NaN;

Answer 3

一般来说，要对矩阵的某些元素执行一项操作，而对其余元素执行另一项操作，一行解决方案是：

Z = B .* X + ~B .* Y;

其中 B 是逻辑矩阵。 例如，

Z = (A == 0) .* -1 + (A ~= 0) .* A;

复制 A，但在 A 为零的地方分配 -1。

然而，因为这个问题涉及无穷大或 NaN，所以可以更简洁地完成：

Z = FAV ./ A; % produces inf where A == 0
Z = (A ~= 0) .* FAV ./ A; % produces NaN where A == 0

Answer 4

您是否正在寻找所有非零元素？ 您可以通过几种方式做到这一点。

nonzero = find(A); % returns indicies to all non-zero elements of A
y = x./A(nonzero); % divides x by all non-zero elements of A
                   % y will be the same size as nonzero

或者对于单行语句，您可以使用条件代替索引

y = x./A(A~=0); % divides x by all non-zero elements of A

Answer 5

您需要做的是确定要操作的元素。 我会使用“查找”。 我将结果存储在 VI（有效索引）中，并使用它来填充矩阵。

clear
clc

den = [2 0 2; 0 2 0; -2 -2 -2]
num = ones(size(den));
frac = nan(size(den));

vi = (den ~=0)

frac(vi) = num(vi)./den(vi)

vi = (den == 0)

frac(vi) = nan %just for good measure...