File: nanmean.m

package info (click to toggle)
dynare 5.3-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 77,852 kB
  • sloc: cpp: 94,481; ansic: 28,551; pascal: 14,532; sh: 5,453; objc: 4,671; yacc: 4,442; makefile: 2,923; lex: 1,612; python: 677; ruby: 469; lisp: 156; xml: 22
file content (69 lines) | stat: -rw-r--r-- 1,998 bytes parent folder | download
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
63
64
65
66
67
68
69
function y = nanmean(x, dim)

% Returns the mean of a matrix with some NaNs.
%
% INPUTS
% - x      [double]    m*n matrix
% - dim    [integer]   scalar, dimension along which the mean has to be computed.
%
% OUTPUTS
% - y      [double]    1*n vector (if dim=1) or m*1 vector (if dim=2).
%
% REMARKS
% (1) Default value for dim is the first non singleton dimension.
% (2) Works with vector and matrices, not implemented for arrays with more
% than two dimensions.

% Copyright (C) 2011-2018 Dynare Team
%
% This file is part of Dynare.
%
% Dynare 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.
%
% Dynare 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 Dynare.  If not, see <https://www.gnu.org/licenses/>.

if nargin<2
    % By default dim is the first non singleton dimension
    nonsingletondims = find(find(size(x)>1));
    if ~isempty(nonsingletondims)
        dim = nonsingletondims(1);
    else
        dim = NaN;
    end
end

switch ndims(x)
  case 1
    if isnan(dim)
        y = x;
    else
        y = mean(x(find(~isnan(x))), dim);
    end
  case 2
    if isnan(dim)
        y = x;
    else
        if isequal(dim, 1)
            y = NaN(1, size(x, 2));
            for i = 1:size(x, 2)
                y(i) = mean(x(find(~isnan(x(:,i))),i));
            end
        else
            y = NaN(size(x, 1), 1);
            for i = 1:size(x, 1)
                y(i) = mean(x(i, find(~isnan(x(i,:)))));
            end
        end
    end
  otherwise
    error('This function is not implemented for arrays with dimension greater than two!')
end