1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62
|
function resu = nth_element(df, n, dim)
%# function resu = nth_element(x, n, dim)
%# This is a wrapper for the real nth_element
%% Copyright (C) 2009-2017 Pascal Dupuis <cdemills@gmail.com>
%%
%% This file is part of the dataframe package for Octave.
%%
%% This package is free software; you can redistribute it and/or
%% modify it under the terms of the GNU General Public
%% License as published by the Free Software Foundation;
%% either version 3, or (at your option) any later version.
%%
%% This package is distributed in the hope that it will be useful,
%% but WITHOUT ANY WARRANTY; without even the implied
%% warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
%% PURPOSE. See the GNU General Public License for more
%% details.
%%
%% You should have received a copy of the GNU General Public
%% License along with this package; see the file COPYING. If not,
%% see <http://www.gnu.org/licenses/>.
if (~isa (df, 'dataframe'))
resu = []; return;
end
if (nargin < 3)
%# default: operates on the first non-singleton dimensio
resu = df.x_cnt;
dim = find (resu > 1);
dim = dim(1);
end
switch dim
case {1}
resu = df_colmeta (df);
for indi = (1:df.x_cnt(2))
resu.x_data{indi} = feval (@nth_element, df.x_data{indi}(:, df.x_rep{indi}), n, dim);
resu.x_rep{indi} = 1:size (resu.x_data{indi}, 2);
end
resu.x_cnt(1) = max (cellfun ('size', resu.x_data, 1));
if (resu.x_cnt(1) == df.x_cnt(1))
%# the func was not contracting
resu.x_ridx = df.x_ridx;
resu.x_name{1} = resu.x_name{1}; resu.x_over{1} = resu.x_over{1};
end
case {2}
error ('Operation not implemented');
case {3}
resu = df_allmeta(df);
for indi = (1:df.x_cnt(2))
resu.x_data{indi} = feval (@nth_element, df.x_data{indi}(:, df.x_rep{indi}), n, dim-1);
resu.x_rep{indi} = 1:size (resu.x_data{indi}, 2);
end
otherwise
error ('Invalid dimension %d', dim);
end
resu = df_thirddim (resu);
end
|