如何將列平均值分配給矩陣的該列中的特定值？

Question

我想有這樣的：

Mu = mean(X); % column-wise means 
X(X == 0) = Mu(current_column); % assign the mean value of the current column 
           % to the zero-element 

但我怎麼告訴MATLAB我要分配的當前列的平均值（即當前列中的零值是）以矩陣項當前的零值？

Answer 1

你可以做一個陣列相同的形狀，包含X列方向是指：

means = repmat(mean(X), [size(X,1) 1]); 
X(X==0) = means(X==0); 

[編輯追加...]

或者，如果陣列冒犯的明確擴張你，你可以這樣做：

X = bsxfun(@(x,y)(x+(x==0)*y), X, mean(X)); 

這是對我的口味有點太「聰明」，但似乎是約25％的速度在我測試的一個案例（1000×1000陣列，約10％是零）。

Answer 2

這是一個矢量化解決方案，它比使用BSXFUN更快，並且不需要複製列方法數組。它只是找到對應的列索引的修改每一個線性指標，然後使用該索引，以獲得正確的列均值：

colMeans = mean(X); %# Get the column means 
index = find(X == 0); %# Get the linear indices of the zero values 
colIndex = ceil(index./size(X,1)); %# Get the column index for each linear index 
X(index) = colMeans(colIndex);  %# Reassign zeroes with the column means 

而這裏的測試用例：

>> X = randi([0 1],5) %# Generate a random matrix of zeroes and ones 

X = 

    0  1  0  1  0 
    1  0  0  1  1 
    0  1  0  1  0 
    1  1  1  0  1 
    0  1  0  0  1 

>> colMeans = mean(X); 
>> index = find(X == 0); 
>> colIndex = ceil(index./size(X,1)); 
>> X(index) = colMeans(colIndex) 

X = 

    0.4000 1.0000 0.2000 1.0000 0.6000 
    1.0000 0.8000 0.2000 1.0000 1.0000 
    0.4000 1.0000 0.2000 1.0000 0.6000 
    1.0000 1.0000 1.0000 0.6000 1.0000 
    0.4000 1.0000 0.2000 0.6000 1.0000