find() 的奇怪行为

Question

我有这个矩阵：

a = [1 2 2 1; 1 1 2 2]

%   1     2     2     1 
%   1     1     2     2

我想找到所有的 1 并将它们归零。

[~, a_i] = find(a == 1);
a(a_i) = 0

%   0     2     2     1 
%   0     0     2     2

为什么第一行还有一个 1？

Answer 1

按照您的操作方式，您只会获得1 的列索引，因为您只使用find 的第二个输出。

[~, col] = find(a == 1)
%   1   1   2   4

当您将其用作a 的索引时，它会将它们视为线性索引，并且仅将a 中的第一个、第二个和第四个值更改为0。线性索引是在column-major order 中执行的，所以这会产生您所看到的输出。

要做你想做的事，你需要find 的both 输出来获取行和列索引，然后使用sub2ind 将它们转换为线性索引然后可以用来索引到a。

[row, col] = find(a == 1);
a(sub2ind(size(a), row, col)) = 0;

使用find 的单一输出版本要容易得多，它直接返回线性索引并使用它。

ind = find(a == 1);
a(ind) = 0;

或者更好的是，只需使用逻辑索引

a(a == 1) = 0;