Matlab arrayfun 使用

http://www.ilovematlab.cn/thread-93830-1-1.html

 

利用arrayfun函数可以避免无谓的循环，从而大大提高代码的简介性。

1、A=ARRAYFUN(FUN, B)
FUN是函数句柄，对B中的每一个元素调用FUN函数(计算顺序随意)，结果存放于A中，size(A)==size(B)
FUN函数可接受numeric, logical, char, struct, cell的自变量类型。
2、[A, B, ...] = ARRAYFUN(FUN, C, ...)
FUN函数的返回值是一个向量，对B中的每一个元素调用FUN函数,计算结果放在A,B…中。
3、 A = ARRAYFUN(FUN, B, C,  ...)
FUN函数接受参数不唯一，分别调用B, C,  ...中的元素，A(i,j...)=Fun(B(i,j...),C(i,j,...)...)，
B,C...大小必须相等。
例1、Fun调用B中的一个数值

1. 
   
2. >> s=[1 2;3 4];
   
3. >> f=@(x) x^2;
   
4. >> arrayfun(f,s)
   
5. ans =
   
6.      1     4
   
7.      9    16
   

复制代码

例2、Fun的接受参数是一个向量

1. 
   
2. >> ss=num2cell(s,2);
   
3. >> f=@(x) sum(x{:}.^2);
   
4. >> arrayfun(f,ss)
   
5. ans =
   
6.      5
   
7.     25
   

复制代码

例3、Fun函数返回值是向量

1. 
   
2. >> F=@(x) x{:}.^2;
   
3. >> cell2mat(arrayfun(F,ss,'UniformOutput',false))
   
4. ans =
   
5.      1     4
   
6.      9    16
   

复制代码

例4、Fun函数参数不唯一

1. 
   
2. >> f=@(x,y) x^2+y^2;
   
3. >> [X,Y]=meshgrid(-2:2,-1:5);
   
4. >> Z=arrayfun(f,X,Y);
   
5. >> mesh(X,Y,Z)