提高Matlab程序运行效率之find函数

如果程序中的循环较多，会导致程序运行效率很低，而find函数则可以解决大部分的循环问题，进而提高程序运行效率。

(1) find函数基本功能

返回向量或者矩阵中不为0的元素的位置索引

X = [1 0 4 -3 0 0 0 8 6];
ind = find(X)

ind =

 1     3     4     8     9

find还有多种用法，如：返回前k个不为0的元素；

ind = find(X,2)

ind =

  1     3

也可以写成

ind = find(X,2,‘first’)

ind =

  1     3

返回后k个不为0的元素

ind = find(X, 2, ‘last’)

ind =

  8    9

若X为一个矩阵，索引应该如何返回呢？

X = [1 -3 0;0 0 8;4 0 6]

X =

 1    -3     0
 0     0     8
 4     0     6

ind = find(X)

ind =

 1
 3
 4
 8
 9

这是因为Matlab在存储矩阵时，是按照列来存储的，e.g.：

X(4)

ans =

-3

假如需要同时输出元素的行和列信息时：

[r,c] = find(X)
r =
3
2
c =
1
3

若还需要输出具体的元素值：

[r,c,v] = find(X)
r =
1
3
1
2
3
c =
1
1
2
3
3
v =
1
4
-3
8
6

(2)进阶技巧

①find()函数的功能是找到向量或者矩阵中不为0的元素，那么若需要找到其中满足一定条件的元素，如，等于4的元素，

X = [1 0 4 -3 0 0 0 8 6];
ind = find(X == 4)

ind =

 3

Matlab中，有一个logical数据类型，和C++中的bool值相同。find()函数将logical值0也视为数值0，logical值1视为数值1，因此可以用上面的方法找到矩阵中满足一定条件的元素。
②程序中，我们经常会以向量中是否包含某一元素为判断条件，比如X中是否有等于9的元素，这时也可以应用find()函数：

if isempty(find(X == 9))
log = 0
else
log = 1
end
log =

注释：当输入中无非0元素时，find（）返回一个空数组。

(3) 高级技巧

在②中，假如向量X的阶数很高，如100000万维，如果继续用以上的方法，则运行速度会很慢。而②中并不需要找到所有等于9的元素，只希望判断是否有等于9的元素，则可对程序做如下改动：

X = randi(100,1,10000000);
tic
for i = 1:100
isempty(find(X == 9));
end
toc

Elapsed time is 1.954537 seconds.

更改后：

X = randi(100,1,10000000);
tic
for i = 1:100
isempty(find(X == 9,1));
end
toc

Elapsed time is 0.757994 seconds.

运行效率约为原来的2.5倍。