取稀疏矩陣每列中所有非零元素的乘積

Question

有沒有一種很好的矢量化方法來獲取八度（或matlab）稀疏矩陣的每列中所有非零元素的乘積（返回一行 - 產品的向量）？

Answer 1

我與accumarray結合find：

%# create a random sparse array 
s = sprand(4,4,0.6); 

%# find the nonzero values 
[rowIdx,colIdx,values] = find(s); 

%# calculate product 
product = accumarray(colIdx,values,[],@prod) 

---

一些替代品（可能是低效率的，你可能要分析他們）

%# simply set the zero-elements to 1, then apply prod 
%# may lead to memory issues 
s(s==0) = 1; 
product = prod(s,1); 

。

%# do "manual" accumarray 
[rowIdx,colIdx,values] = find(s); 

product = zeros(1,size(s,2)); 
uCols = unique(colIdx); 

for col = uCols(:)' 
    product(col) = prod(values(colIdx==col)); 
end 

Answer 2

我找到了一種替代的方法來解決這一點，但它可能是速度較慢，不太在最壞的情況下精確：

簡單地採取所有非零元素的日誌，然後求和列。然後取出結果向量的exp：

function [r] = prodnz(m) 
    nzinds = find(m != 0); 
    vals = full(m(nzinds)); 
    vals = log(vals); 
    m(nzinds) = vals; 
    s = full(sum(m)); 
    r = exp(s); 
endfunction