MATLAB可变参数函数

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
则上面的函数,在调用时,提供至少2个参数,否则会报错。

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

  • 5
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值