nansum.m

来自「不完全数据分析MATLAB程序（部分信息重建）：最小均方估计、协方差矩阵、缺失值」· M 代码 · 共 24 行

function y = nansum(x)
%NANSUM   Sum ignoring NaNs.
%
%    NANSUM(X) returns the sum over non-NaN elements of X.  For
%    vectors, NANSUM(X) is the sum of the non-NaN elements in X. For
%    matrices, NANSUM(X) is a row vector containing the sum of the
%    non-NaN elements in each column of X.
%
%    See also NANMEAN, NANSTD.

  error(nargchk(1,1,nargin))          % check number of input arguments 
				      
  % replace NaNs with zeros.
  nans    = isnan(x);
  inan    = find(nans);
  x(inan) = zeros(size(inan));

  y       = sum(x);
  
  % protect against an entire column of NaNs
  iall    = find(all(nans));
  y(iall) = NaN;