заполнение NaNs на основе минимального числового значения

Question

У меня есть матрица A, и я хотел бы заполнить NaN в каждом столбце до точки, где встречается числовое значение.

Пример

A = 
    [ 0 0 0 
     0 0 6 
     5 7 0] 

должна быть

A = 
    [NaN NaN NaN 
     NaN NaN 6 
     5 7 0] 

Есть ли способ векторизации этой проблемы, а не делать это в два для петли шага

Спасибо за вашу помощь!

Answer 1

Использование max найти первый ненулевой (если таковые имеются) в каждом столбце, а затем bsxfun, чтобы создать маску для значений, которые должны быть установлены в NaN:

[valid, row] = max(A~=0); 
row(~valid) = size(A,1)+1; %// in case some column contains all zeros 
A(bsxfun(@lt, (1:size(A,1)).', row)) = NaN;