如何根据输入参数的数量动态创建嵌套循环

时间:2018-05-14 16:03:47

标签: matlab function loops recursion nested-loops

我正在尝试使用函数绘制数据表,其中每个系列都被视为一组特定标准的数字,如下所示:

function plotter_fun(Table)

    aList = unique(Table.a);
    bList = unique(Table.b);
    cList = unique(Table.c);

    for a = aList
        for b = bList
            for c = cList
                Slice = Table(Table.a==a & Table.b==b & Table.c==c, :);
                plot(Slice.x, Slice.y);
            end
        end
    end
end

我想这样做,以便表头参数('a','b','c')可以作为参数传入,并且可以是任意数量的参数。期望的结果将是:

function plotter_fun(Table, headerNames)

    for i = 1:numel(headerNames)
        loopList{i} = unique(Table.(headerNames{i}));
    end

    % do nested looping

end

它被称为:

plotter_fun(Table, {'a', 'b', 'c'});

我不确定如何使用递归循环算法,以便可以动态更改嵌套循环的数量?

1 个答案:

答案 0 :(得分:2)

看起来每个切片都是您的标头变量的唯一组合'值。考虑到这一点,我们可以使用findgroupsunique完全消除循环。那么"动态嵌套"情况不是问题...

此功能的工作原理如下:

function plotter_fun(Table, headerNames)
    % Create a matrix of group indices for each specified header
    grps = zeros(size(Table,1), numel(headerNames));
    for i = 1:numel(headerNames)
        grps(:,i) = findgroups(Table.(headerNames{i}));
    end
    % Get the unique rows for the grouping variables
    [~, ~, idx] = unique(grps, 'rows');
    % Loop over each unique row index and slice out the specified rows.
    for i = 1:max(idx)
        Slice = Table( idx == i, : );
        plot( Slice.x, Slice.y );
    end
end

测试(当删除plot行时,这会有效,因为我还没有指定xy列:

tbl = cell2table( {'a', 1, 'dog'; 
                   'a', 2, 'cat'; 
                   'b', 3, 'cat'; 
                   'a', 2, 'cat'}, ...
                 'variablenames', {'char','num','pet'} )

plotter_fun( tbl, {'char', 'num', 'pet'} ) % output slices are rows [1], [2,4] and [3].

plotter_fun( tbl, {'char'} ) % output slices are rows [1,2,4] and [3].

编辑:

这是一种灵活的方式来自动生成"过滤器"来自Slice的标签。我们可以在=和表值之间连接headerNames(如果有数值,则使用num2str转换),然后使用strjoin生成由{{,分隔的标签每个标题1}}。

一个衬里看起来像这样,并将在定义Slice的循环中使用:

label = strjoin( strcat( headerNames', '=', ...
                   cellfun(@num2str,table2cell(Slice(1,headerNames)),'uni',0)' ), ', ');

% Output for Slice 1 of example 1 above: 'char=a, num=1, pet=dog'
% Output for Slice 1 of example 2 above: 'char=a'

% cellfun(@num2str, __ ) conversion not needed if there is no numeric data in the table