Python中yield关键字的Matlab等价物是什么?

时间:2014-01-13 18:51:53

标签: python matlab generator yield yield-keyword

我需要一次生成多个结果,而不是一次生成数组中的所有结果。

如何在Matlab中使用像Python一样的语法生成器?

2 个答案:

答案 0 :(得分:10)

执行使用yield关键字的函数时,它们实际上会返回一个生成器。生成器是一种迭代器。虽然MATLAB没有提供任何一种语法,但您可以自己实现"iterator interface"。这是一个类似于python中的xrange函数的示例:

classdef rangeIterator < handle
    properties (Access = private)
        i
        n
    end

    methods
        function obj = rangeIterator(n)
            obj.i = 0;
            obj.n = n;
        end

        function val = next(obj)
            if obj.i < obj.n
                val = obj.i;
                obj.i = obj.i + 1;
            else
                error('Iterator:StopIteration', 'Stop iteration')
            end
        end

        function reset(obj)
            obj.i = 0;
        end
    end
end

以下是我们如何使用迭代器:

r = rangeIterator(10);
try
    % keep call next() method until it throws StopIteration
    while true
        x = r.next();
        disp(x);
    end
catch ME
    % if it is not the "stop iteration" exception, rethrow it as an error
    if ~strcmp(ME.identifier,'Iterator:StopIteration')
        rethrow(ME);
    end
end

注意在迭代器上使用Python中的构造for .. in ..时,它在内部做了类似的事情。

您可以使用常规函数而不是类来编写类似的东西,使用persistent变量或闭包来存储函数的本地状态,并在每次调用时返回“中间结果”。

答案 1 :(得分:2)

在MATLAB中(还没有?在Octave中),您可以使用闭包(嵌套,范围函数):

function iterator = MyTimeStampedValues(values)

    index = 1;

    function [value, timestamp, done] = next()
        if index <= length(values)
            value = values(index);
            timestamp = datestr(now);
            done = (index == length(values));
            index = index + 1;
        else
            error('Values exhausted');
        end
    end

    iterator = @next;
end

然后

iterator = MyTimeStampedValues([1 2 3 4 5]);
[v, ts, done] = iterator();    % [1, '13-Jan-2014 23:30:45', false]
[v, ts, done] = iterator();    % ...