上一篇博客介绍了通过向量化运算、预分配空间两种常用的方式提高代码的运行效率。实际上,matlab还有几种内置函数来避免循环语句的使用,分别为arrayfun、cellfun以及structfun函数。由于这几种函数需要用到匿名函数以及函数句柄的概念,很多人可能不太清楚,所以我才单独写一篇博客进行介绍。
1.匿名函数与函数句柄
匿名函数是 Matlab 中的一种函数类型,它可以方便地定义和使用轻量级的函数,通常被用于一些简单的计算和数据分析任务中。匿名函数的语法简单,能够大大提高代码的可读性和可维护性,也更加符合函数式编程的思想。
1.1匿名函数的定义
在 Matlab 中,定义匿名函数的语法如下:
f = @(arguments) expression
其中,f 表示匿名函数名称(也称为函数句柄),arguments 为函数中的变量,expression 为匿名函数的表达式。参数列表可以是零个或多个参数,多个参数使用逗号分隔。表达式返回值将作为函数的输出。例如,可以定义一个匿名函数,功能是将输入的变量 x 先平方再加1:
f = @(x) x.^2 + 1;
这个匿名函数的表达式是 x.^2 + 1,意思是对输入参数 x 逐元素平方并加1,并输出一个新的数组。另外,和普通的子函数一样,也可以定义没有参数的匿名函数,例如:
g = @() disp('Hello, world!');
这个匿名函数不需要输入参数,它的表达式是 disp(‘Hello, world!’),也就是在命令行输出“Hello, world! ”字符串。
1.2匿名函数的使用
定义完匿名函数之后,我们可以通过函数句柄来调用这个匿名函数。调用匿名函数的语法与调用普通函数一样,例如:
f = @(x) x.^2 + 1; x=1:10; y=f(x)
输出结果为:
1.3函数句柄的定义
刚才也提到了,定义匿名函数,匿名函数的名称也叫做函数句柄。其实,在 Matlab 中,通过在函数名前面加上 “@” 符号,就可以定义一个函数句柄。而函数句柄的使用方式类似于函数名的使用方式。如果有C或者C++的基础,可以发现所谓的函数句柄其实和指针的概念非常相似。函数句柄即可以指向普通的子函数,也可以指向匿名函数,例如:
f1=@fun1; % 定义函数句柄f1指向函数fun1 f2=@(x) x.^2; % 定义函数句柄f2指向匿名函数 a=[1 2 3]; b1=f1(a) b2=f2(a) function y = fun1(x) y = x.^2; end
输出结果完全一样
1.4函数句柄的使用
使用函数句柄,我们还能够实现动态函数调用。例如,在 Matlab 中,如果我们想要调用内置函数或者自定义的子函数,可以使用这些函数的函数句柄来实现动态调用,而不需要显式地调用这些函数。这么说可能比较难理解,举个简单的例子:
f1=@fun1; % 定义函数句柄f1指向函数fun1 f2=@fun2; % 定义函数句柄f2指向函数fun2 f3=@fun3; % 定义函数句柄f3指向函数fun3 f4=@fun4; % 定义函数句柄f4指向函数fun4 a=[1 2 3]; b1=fun(f1,a)+fun(f2,a)+fun(f3,a)+fun(f4,a) b2=f1(a)+f2(a)+f3(a)+f4(a) function y = fun1(x) y = x; end function y = fun2(x) y = 2*x.^2; end function y = fun3(x) y = 3*x.^3; end function y = fun4(x) y = 4*x.^4; end function y = fun(f,x) y = f(x); end
这个例子里,我们首先定义了子函数fun1-fun4,用于计算k*x^k。还定义了子函数fun,这个函数的作用是传入一个函数句柄f和一个变量x,输出使用传入的函数f对变量x处理后的结果y。b1和b2的结果是完全一样的:
通过这个例子,大家可能体会到了函数句柄进行动态函数调用的用法,但是并没有感觉使用函数句柄进行动态函数调用有哪些优势,我将用另外一个例子进行说明。
假设我们需要一个数据处理的程序,用于对输入的数据进行多个不同的数据转换操作,例如对数据进行过滤、自动缩放、归一化等操作。为了方便,我们希望将这些不同的数据转换操作都封装成具体的函数,并使用一个统一的函数来调用这些函数。使用函数句柄,我们能够实现这一点。例如:
clc clear x = -5:0.1:5; % 定义原始输入数据 y1 = processData(x, 'filterData'); % 过滤数据 y2 = processData(x, 'scaleData'); % 缩放数据 y3 = processData(x, 'normalizeData'); % 数据归一化 % 绘制计算结果 subplot(2,2,1); plot(x,x); subplot(2,2,2); plot(x,y1); subplot(2,2,3); plot(x,y2); subplot(2,2,4); plot(x,y3); % 定义具体的数据转换函数 function y1 = filterData(x) y1(x>0) = x(x>0); y1(x<=0) = 0; end function y2 = scaleData(x) y2 = x./max(abs(x)); end function y3 = normalizeData(x) y3 = (x-mean(x))./std(x); end % 定义数据转换调用函数 function y = processData(x, funcName) func = str2func(funcName); % 将函数名转换为函数句柄 y = func(x); % 根据函数句柄调用具体的数据转换函数 end
运行结果为:
如果经常使用Matlab应该会发现,Matlab的内置函数也有很多这样的函数,通过类似下面语句完成函数处理,可以大大提升效率,这里我不再过多介绍。
函数名(变量,'函数关键字')
2.通过内置函数arrayfun、cellfun、structfun避免使用循环
在Matlab中,arrayfun、cellfun、structfun是三个用于处理数组、单元数组和结构体的函数,它们可以帮助用户快速进行数据处理。
2.1.arrayfun函数
arrayfun函数是用于对数组进行批量处理的函数,它的基本语法如下:
B = arrayfun(fun, A)
其中,第一个参数 fun 是一个函数句柄,表示对 A 中每个元素进行的操作,返回的结果保存在一个新的数组 B 中。arrayfun函数将逐个处理 A 中的每个元素,并将结果传递给 fun 函数进行处理。
举个例子,假设我们要对一个向量 x 中的每个元素进行平方处理,可以使用 arrayfun 来实现:
x = [1 2 3 4 5]; y = arrayfun(@(a) a^2, x); disp(y);
在这个例子中,我们定义了一个匿名函数 @(a) a^2,并将它作为 arrayfun 的第一个参数传入。arrayfun 将逐个处理 x 中的每个元素,并将结果传递给匿名函数进行平方处理。最终,arrayfun 将这些结果存储在另一个向量 y 中,并输出。
2.2 cellfun函数
cellfun函数是一个用于对单元数组进行批量处理的函数,它的基本语法如下:
B = cellfun(fun, C)
其中,第一个参数 fun 是一个函数句柄,表示对 C 中每个元素进行的操作,返回的结果保存在一个新的单元数组 B 中。cellfun函数将逐个处理 C 中的每个元素,并将结果传递给 fun 函数进行处理。
举个例子,假设我们要将一个单元数组 C 中每个元素的大小写转换,可以使用 cellfun 来实现:
C = {'Hello', 'World'}; D = cellfun(@(a) upper(a), C, 'UniformOutput', false); disp(D);
在这个例子中,我们定义了一个匿名函数 @(a) upper(a),表示将一个字符串转换成大写。我们将匿名函数作为 cellfun 的第一个参数进行调用,cellfun 将逐个处理 C 中的每个元素,并将结果传递给匿名函数进行大小写转换。最终,cellfun 将这些结果存储在另一个单元数组 D 中,并输出。需要注意的是,在使用 cellfun 时,需要指定参数 'UniformOutput' 为 false,否则函数的返回结果不会被封装为单元数组,因为不同大小的返回值无法合并为单个单元数组。
2.3structfun函数
structfun函数是用于对结构体进行批量处理的函数,它的基本语法如下:
B = structfun(fun, S)
其中,第一个参数 fun 是一个函数句柄,表示对 S 中每个字段进行的操作,返回的结果保存在一个新的结构体 B 中。structfun将逐个处理 S 中的每个字段,并将结果传递给 fun 函数进行处理。
举个例子,假设我们要将一个结构体 S 中的数值型字段乘以2,可以使用 structfun 来实现:
S.x = 1; S.y = 2.5; S.z = 3.7; T = structfun(@(a) 2*a, S, 'UniformOutput', false); disp(T);
在这个例子中,我们定义了一个匿名函数 @(a) 2*a,表示将一个字段的值乘以2。我们将匿名函数作为 structfun 的第一个参数进行调用,structfun 将逐个处理 S 中的每个字段,并将结果传递给匿名函数进行处理。
2.4 避免循环语句的使用
使用 cellfun 可以以简洁的方式避免使用循环语句,并提高代码的可读性和可维护性。下面我们通过一个实例来说明。
例1:单元数组中字符串拼接
假设我们有一个单元数组 C,其中包含了字符串,我们想要将其中的字符串进行拼接,并返回一个新的字符串。我们先使用循环语句来实现,代码如下:
% 定义数组 A = {{'H','e','l','l','o',' '}, {'W','o','r','l','d'}, {'——','M','a','t','l','a','b'}}; B = cell(1,10000); for k=1:10000 B{k}=A; end % 循环计算 result = ''; tic for k = 1:numel(B) for kk=1:numel(B{k}) for kkk=1:numel(B{k}{kk}) result = [result, B{k}{kk}{kkk}]; end end end toc
运行时间为:
这段代码使用了循环语句来计算单元数组中字符串的拼接,使用一个 for 循环来遍历单元数组中的元素,然后每次循环将当前元素添加到最终结果中,如果使用 cellfun 函数来实现相同的功能,代码如下:
可以使用如下`cellfun`函数来避免循环:```
A = {{'H','e','l','l','o',' '}, {'W','o','r','l','d'}, {'——','M','a','t','l','a','b'}}; B = cell(1,10000); for k=1:10000 B{k}=A; end tic C = cellfun(@(x) cellfun(@(y) [y{:}], x, 'UniformOutput', false), B, 'UniformOutput', false); D = [C{:}]; result = [D{:}]; toc
上述代码中,使用了两个嵌套的`cellfun`函数,将多维cell数组中的字符元素连接成一个字符串。其中第一个`cellfun`函数会对`B`中每个元素应用第二个`cellfun`函数,将其分别处理成一个字符串序列并封装到一个cell中。最后使用`{:}`操作符将所有结果取出来,连接成一个字符串。这样,就可以避免使用for循环,提高了代码运行的效率。