1. 概述
在使用MATLAB的时候,可以发现,同样的函数,会因为接收的参数个数不同,或者返回参数的个数不同,导致输出的内容也不相同,例如:
A=ones(4);
B=ones(3,4);
其结果是A为4x4的矩阵,而B是3x4的矩阵。
类似的,进行奇异值分解的svd函数,常用的方式如下:
S=svd(A)
[U,S,V] = svd(A)
其中S是返回的奇异值,U和V分别是左/右奇异向量。
这种行为类似于其他编程语言中的函数重载,又有些类似函数缺省参数的方式,不过深究起来,MATLAB的函数实现方式还是和其他的语言有很大的不同,我们也可以通过MATLAB提供的一些参数实现类似的功能。
2. 可变参数输入
varargin 是函数定义语句中的一个输入变量,允许函数接受任意数量的输入参数。使用小写字符指定 varargin。在任何显式声明的输入项后,附加 varargin 作为最后一个输入参数。
在执行函数时,varargin 是一个 1×N 元胞数组,其中 N 是函数在显式声明的输入后收到的输入项数。如果该函数在显式声明的输入后未收到任何输入,则 varargin 是空元胞数组。
例如,要对一个计算未知数量个数的值求和(当然,MATLAB的sum函数可以做的更好,但是这次主要用于解释可变参数的使用)。可以进行如下定义:
function y=xsum(varargin)
disp("Number of input arguments: " + nargin);
y=0;
for i=1:length(varargin)
y=y+varargin{i};
end
在使用的时候,由于varargin是元胞数组,所以需要用花括号{}提取其中的元素。第二行中的nargin 是值输入参数的个数。
在函数中varargin和nargin是只读的,它们在函数调用的时候就会被确定。
另外,就是有些输入是有强制最小数量的输入的,例如我们要定义一个求2个或多个数中的最大值,可以这样定义:
function y=xmax(a,b,varargin)
if nargin<2
error("参数的数目不足");
end
y=a;
if y<b
y=b;
end
for i=1:length(varargin)
if y<varargin{i}
y=varargin{i};
end
end
3. 可变参数输出
与varargin和nargin对应的,还有varargout和nargout参数,用于实现可变数目的输出。同样的,nargout也是在函数调用的时候就会被确定,不过和varargin只读不同,varargout是可读写的。例如,我们采用二分法,对方程求数值根,如果直接调用,或者提供一个参数,则返回的是函数的根,如果提供的是两个参数,则第二个参数返回的是迭代次数,可编写函数如下:
%%
% 使用二分法求方程的根
% func:待求根的函数句柄
% a,b:求根范围
% err:误差精度
%%
function [root,varargout]=bisection(func,a,b,err)
count=0;
MAX_COUNT=100;
if sign(func(a))*sign(func(b))>0
error("bad range");
end
while count<MAX_COUNT
c=(a+b)/2;
root=func(c);
count=count+1;
if abs(root)<err
break;
end
if sign(root)==sign(func(a))
a=c;
else
b=c;
end
end
root=c;
if nargout==2
varargout{1}=count;
end
调用示例:
>> f=@(x) exp(x)-2-cos(exp(x)-2);
>> bisection(f,0.5,1.5,1e-10)
ans =
1.007623971672729
>> [root,cnt]=bisection(f,0.5,1.5,1e-10)
root =
1.007623971672729
cnt =
32