取稀疏矩阵每列中所有非零元素的乘积

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