向量化编程:arrayfun函数的使用

rocwoods

最近发现的一个技巧，台湾去年就有人用了，在simwe、研学、振动上搜了搜都没有这个函数的介绍。因此特转过来。
我们知道，MATLAB编程核心思想之一就是向量化。MATLAB的很多built-in函数，向sin,cos,find等等都支持向量运算。但是，很多时候我们编写的函数的输入变量是标量，而我们又要对很多组参数进行函数调用，MATLAB7以前的版本中我们只能通过循环来实现，如果参数的维数增加，就会出现循环套循环的现象，效率恐怖。从MATLAB7.1开始，MATLAB新增arrayfun这个built-in函数来实现将任意函数应用到数组内包括结构在内的所有元素。这样很多以前不可避免的循环现在可以向量化了。举例如下：
      例1：生成一个这样的n*n矩阵a：a(i,j)=dblquad(@(u,v) sin(u)*sqrt(v),0,i,0,j)。以n=10为例，
      以前我们可能这样做：

1. a=zeros(10);
   
2. for ii=1:10
   
3.   for jj=1:10
   
4.      a(ii,jj)=dblquad(@(u,v) sin(u)*sqrt(v),0,ii,0,jj);
   
5. end
   
6. end

复制代码

现在我们只需这样：

1. [J,I]=meshgrid(1:10);
   
2. a1=arrayfun(@(ii,jj) dblquad(@(u,v) sin(u)*sqrt(v),0,ii,0,jj),I,J);

复制代码

例2：验证角谷猜想，一个正整数n,如果是偶数除以2，如果是奇数乘以3加1，得到的新数继续按上述规则运算，最后结果都为1。验证1到100000内的正整数。
先编写单个数的验证函数

1. function f=jiaogu(n)
   
2. while(n>1)
   
3. if (mod(n,2)==1)
   
4.     n=n*3+1;
   
5. elseif(mod(n,2)==0)
   
6.     n=n/2;
   
7. else break;
   
8. end
   
9. end
   
10. f=n;

复制代码

验证1:100000, 以前我们可能这样做：

1. a=zeros(1,100000);
   
2. for k=1:100000
   
3. a(k)=jiaogu(k);
   
4. end
   
5. all(a)

复制代码

现在我们只需这样：

1. all(arrayfun(@jiaogu,1:100000))

复制代码

举这两个例子目的是为了让大家知道很多以前不能优化的程序利用这个函数还可以进一步优化。具体了解请doc arrayfun 。另外cellfun和structfun也提供了类似的功能，关于其用法，帮助文档里写的很详细。

[ 本帖最后由 rocwoods 于 2007-8-13 11:22 编辑 ]

TurtleRock

看到你在simwe上发过，相当不错，以前对矩阵积分时都用循环，看了这个，感觉会相当方便了。