[MATLAB] find()函数

在做吴恩达机器学习编程作业2的时候，发现在画训练集离散图的时候用到了find() 函数来寻找 y=1或 y=0的元素位置，下面了解一下find函数。
find函数相当于一个简单的loop，它的基本功能是返回向量或者矩阵中不为0的元素的位置索引。
首先看一下函数的参数和返回值：

ind = find(X)
ind = find(X, k) # 返回前k个不为零的元素索引值
ind = find(X, k, 'first') # 返回前k个不为零的元素索引值
ind = find(X, k, 'last') # 返回后k个不为零的元素索引值
[row,col] = find(X, ...) # 返回矩阵X中非零元素的行和列的索引值
[row,col,v] = find(X, ...) # 返回矩阵X中非零元素的行和列的索引值以及该不为零元素

首先在向量中举例：

>> A = [2;0; 4; 0; 3; 5; 6; 0; 8; 0; 0; 1]

A =

     2
     0
     4
     0
     3
     5
     6
     0
     8
     0
     0
     1

>> index = find(A)

index =

     1
     3
     5
     6
     7
     9
    12
%  如果X是一个行向量，则index是一个行向量；否则，index是一个列向量。

可以看到index返回的就是向量中不为0的元素的位置了。
下面介绍在矩阵中寻找不为零的元素位置，要注意一点，matlab中矩阵存储是按列存储的，也就是首先存储第一列的所有值，然后再存储第二列、第三列……如果还不清楚的话，就看一下下面的例子：

>> B = reshape(A(:),4,3)

B =

     2     3     8
     0     5     0
     4     6     0
     0     0     1

>> index = find(B)

index =

     1
     3
     5
     6
     7
     9
    12

想返回这些元素所在的行和列就可以用 [row,col] = find(X, ...)，我这里举例返回前四个不为零的元素的行列值。（按列的顺序排序）

>> [r,c] = find(B,4)

r =

     1
     3
     1
     2


c =

     1
     1
     2
     2

接下来显示我想输出不为零的数值，使用 [row,col,v] = find(X, ...)

>> [r,c,v] = find(B,4)

r =

     1
     3
     1
     2


c =

     1
     1
     2
     2


v =

     2
     4
     3
     5

还可以通过限定参数范围寻找符合条件的数值的索引值。
举例：我想要分别得到该矩阵中数值为5和0的索引值


>> [r,c] = find(B==0)

r =

     2
     4
     4
     2
     3


c =

     1
     1
     2
     3
     3

>> index = find(B == 5)

index =

     6

>>

除此之外find函数还可用在判断语句中，用来判断某向量是否包含某个数值，如果包含就进行下面的步骤。
例如有一场知识竞赛，一共5道题，只要答对一道题就可以进入下一轮，判断是否能进入下一轮比赛。这里使用find就很方便。

>> y = [1 0 0 0 0]；
>>  if ~isempty(find(y == 1))
      fprintf("pass!")
   else
      fprintf("game over!")
end
pass!>>

[MATLAB] find()函数

推荐阅读更多精彩内容