MATLAB for循环优化三例

　　最近一周，对MATLAB有进行了新一轮的学习，对其矩阵化编程的思维有了更深入的了解。确实精妙！

例1：

　　将矩阵A= [1 2 3; 2 4 3; 3 4 5]中所有的数字3替换为33.

如果还停留在C语言的思维可能就是

>> for i=1:3
for j=1:3
if A(i,j)==3
A(i,j) = 33;
end
end
end
>> A
A =
     1     2    33
     2     4    33
    33     4     5

　　而MATLAB语言则有更灵活的语法，

>> A(A==3) = 33
A =
     1     2    33
     2     4    33
    33     4     5

　　这句的意思就是讲matlab中等于3的数复制为33；

>> A==3
ans =
     0     0     1
     0     0     1
     1     0     0

　　当在MATLAB中输入A==3

>> A==3
ans =
     0     0     1
     0     0     1
     1     0     0

　　得到的是一个逻辑数组，接下来的复制操作只对逻辑为真的部分进行。显然后一种写法代码更加简洁。

例2

将A= [1 2;2 3; 3 4; 4 5]每行视为坐标；将5阶零矩阵的这四个位置复制为1；

　　继续常规思维，也是最正常的思维就是每个位置赋值：

>> A= [1 2;2 3; 3 4; 4 5];
B = zeros(5);
for i=1:4
    B(A(i,1),A(i,2)) = 1;
end
>> B
B =
     0     1     0     0     0
     0     0     1     0     0
     0     0     0     1     0
     0     0     0     0     1
     0     0     0     0     0

　　这里可以使用sub2ind这个函数，从字面意思已经知道，是将下标转换为索引值，具体原理可以help sub2ind以及type sub2ind(查看源码)

所以可以直接写成

>> A= [1 2;2 3; 3 4; 4 5];
>> B = zeros(5);
>> B(sub2ind(size(B), A(:,1), A(:,2)))=1
B =
     0     1     0     0     0
     0     0     1     0     0
     0     0     0     1     0
     0     0     0     0     1
     0     0     0     0     0

　　深刻体会到矩阵处理问题的简洁之美！

例3

统计矩阵A = [1 3 4; 5 3 2; 4 2 1]中3的个数

如果还停在初学C语言的思维，恐怕又是靠for解决问题

>> A = [1 3 4; 5 3 2; 4 2 1];
>> a = 0;
>> for i=1:3
for j=1:3
if A(i,j)==3
a = a+1;
end
end
end
>> a
a =
     2

　　这使如果知道find函数和length函数，就会轻松不少：

>> length(find(A==3))
ans =
     2

　总结，通过以上三个例子，可以看到：

　　1、学习一门新的东西，一定要把握其精髓，不要固定步自封（比如一味使用for循环）、生搬硬套；

　　2、就我现在体会，学习matlab最重要的是学习它丰富的函数和灵活的语法，也就是说这门语言处理运算的思维。

　　3、作为一个技术人员，一定要偷懒，抠门。什么是偷懒，相同的功能，能用一行代码解决绝不用两行，什么是抠门，用最简洁的思维，最快的时间、最小的内存。

　　4、敢想敢做，敢想：为什么不能用一行代码解决；敢做，敢于通过学习实现自己的想法。

2015-12-13