如何直接将函数的多个输出传递给另一个?

时间:2016-09-06 23:55:20

标签: matlab function arguments octave

让我详细说明示例:我们知道如何轻松地将功能与单个输出结合起来:

a = sin(sqrt(8));

现在考虑这个示例代码,其中包含两个步骤来计算RXY作为中间输出。

[X, Y] = meshgrid(-2:2, -2:2);
[~, R] = cart2pol(X, Y);

一般有没有办法将这两个函数结合起来并摆脱中间输出?例如,我如何编写与前一代码相同的[~, R] = cart2pol(meshgrid(-2:2, -2:2))类似内容?

注意:我的问题与this question的不同之处在于,在我的情况下,外部函数接受多个输入。因此,我不能也不想将第一个函数的输出组合成一个单元格数组。我希望它们分别传递给第二个函数。

1 个答案:

答案 0 :(得分:2)

要回答标题中的问题:使用以下函数,可以将函数的多个输出重定向到另一个:

function varargout = redirect(source, destination, n_output_source, which_redirect_from_source, varargin)
%(redirect(source, destination, n_output_source, which_redirect_from_source,....)
%redirect output of a function (source) to input of another function(destination)
% source: function pointer
% destination: function pointer
% n_output_source: number of outputs of source function (to select best overload function)
% which_redirect_from_source: indices of outputs to be redirected
% varargin arguments to source function
    output = cell(1, n_output_source);
    [output{:}] = source(varargin{:});
    varargout = cell(1, max(nargout,1));
    [varargout{:}] = destination(output{which_redirect_from_source});
end

现在我们可以将它应用于示例:

[~,R] = redirect(@meshgrid,@cart2pol, 2, [1, 2], -2:2, -2:2)

这里,源函数有2个输出,我们想要将输出1和2从源重定向到目标。 -2:2是源函数的输入参数。

处理上述示例的其他方法:如果您可以使用GNU Octave,bsxfunnthargout这是您的解决方案:

R = bsxfun(@(a,b) nthargout(2, @cart2pol,a,b),(-2:2)',(-2:2))

在Matlab中可能的解决方案是:

[~, R] = cart2pol(meshgrid(-2:2), transpose(meshgrid(-2:2)))

function R = cart2pol2(m)
    [~,R] = cart2pol(m,m')
end

cart2pol2(meshgrid(-2:2, -2:2))
相关问题