在做吴恩达机器学习编程作业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!>>