查看单个帖子
旧 2019-12-10, 16:49   #1
poster
高级会员
 
注册日期: 2019-11-21
帖子: 3,006
声望力: 66
poster 正向着好的方向发展
帖子 矩阵加法跳过NaN值

我想在跳过NaN值的同时对矩阵进行元素明智加法。 MATLAB和Octave具有nansum ,但它在矩阵内按列进行加法运算。

让:

a = NaN * zeros(3) b = ones(3)
我想要:

c = nan+(a, b) 和

c = b
和:

d = nan+(a,a) 和

d = a
回答:
如果沿第n + 1个维度对nd数组进行分类,则仍然可以使用nansum。

对于2D

% commands de-nested for readability. You can do this with a single line, of course tmp = cat(3,a,b); c = nansum(tmp,3); 一般情况

function out = nansumByElement(A,B) %NANSUMBYELEMENT performs an element-wise nansum on the nD arrays A and B % A and B have to have the same size % test input if nargin < 2 || isempty(A) || isempty(B) || ndims(A)~=ndims(B) || ~all(size(A)==size(B)) error('please pass two non-empty arrays of the same size to nansumByElement') end % calculate output nd = ndims(A); % get number of dimensions % catenate and sum along n+1st dimension out = nansum(cat(nd+1,A,B),nd+1);

更多&回答...
poster 当前离线   回复时引用此帖