稀疏矩陣每一行中非零元素的平均值

Question

在以下稀疏矩陣中：

A=[1 1 1 3];
C = sparse(A',1:length(A),ones(length(A),1),4,4);
C =

   (1,1)        1
   (1,2)        1
   (1,3)        1
   (3,4)        1

>>full(C)

ans =

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

如何計算每行中非零元素的平均值？ 我不能在這些稀疏矩陣上使用 matlab 的內置均值 function。 我發現了這個類似的問題，我可以將其應用於我的問題

[row, ~, v] = find(C);
K>> rowmean = accumarray(row, v, [], @mean);
K>> rowmean 

rowmean =

     1
     0
     1

但是，我想為最后一行獲得零值，而不是從答案中刪除這一行。

Answer 1

方法一

您可以指定 output 大小作為accumarray的第三個輸入：

[row, ~, v] = find(C);
rowmean = accumarray(row, v, [size(C,1), 1], @mean);

如果需要，您可以使用accumarray的第六個輸入來獲得sparse output：

rowmean = accumarray(row, v, [size(C,1), 1], @mean, 0, true);

方法二

你可以做一些像這樣更簡單的事情。 結果是sparse的：

rowmean = sum(C, 2) ./ sum(C~=0, 2); % mean of nonzeros in each row, manually
rowmean(isnan(rowmean)) = 0; % replace NaN (resulting from 0/0) by 0