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
|
% Compute the mean (ignoring NaNs).
% s = mean (X, DIM)
% Compute the mean along dimension DIM.
function [s,total,count] = nanmean(self,varargin)
% s will be a cell element containing
% {the sum, the count of elements different from NaN}
% this is necessary to avoid 2 calls to reduce
[s,n] = reduce(self,...
@(x,y) funred(x,y),...
@(x) funelem(x),...
varargin{:});
if isempty(s)
s = 0;
else
total = s{1};
count = s{2};
s = NaN*zeros(size(total));
nonz = count ~= 0;
s(nonz) = total(nonz) ./ count(nonz);
end
end
function x = funelem(x)
% make sure x is not an ncArray
x = full(x);
m = isnan(x);
x(m) = 0;
x = {x,~m};
end
function s = funred(x,y)
s = {x{1} + y{1}, x{2} + y{2}} ;
end
% Copyright (C) 2013 Alexander Barth <barth.alexander@gmail.com>
%
% This program 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 of the License, or
% (at your option) any later version.
%
% This program 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 program; If not, see <http://www.gnu.org/licenses/>.
|