我有一个索引数组 I 和值 X,并且想要创建一个元胞数组 C,以便 C{i} = X(I==i)。计算 C 的最快且最好的方法是什么?
最直接的方法就是评估C{i} = X(I==i)
对于 I 中所有唯一的 i(方法 1):
for i = unique(I)
C{i} = X(I == i);
end
另一种简单的方法是循环 I 中的所有 i 并将相应的 x 附加到 C(方法 2):
C = cellfun(@(x)(zeros(1,0)),cell(1,max(indices)),'UniformOutput',false);
for j = 1:length(I)
i = I(j);
C{i} = cat(2,C{i},X(j));
end
这两种方法都不是很快。为了进行基准测试,我们生成一些测试数据:
I = floor(rand(1,N)*M)+1;
X = rand(1,N);
With N = 1000000, M = 1000
这两种方法采取:
在这里,方法 1 是最好的(仍然很慢)。将问题的参数更改为N = 1000000, M = 10000
显着改变事情:
基本上,这两种方法都太慢了几个数量级。评估 C 的最佳方法是什么?
Edit:正确答案显然是下面乔纳斯的。我附上基准测试结果以供参考。与上述方法相比,C 中元素的顺序有所不同。除此之外,以下给出相同的输出:
C = accumarray(I',X,[],@(x){x'})';
-
N = 100000, M = 1000
:0.0397 秒
-
N = 100000, M = 10000
:0.145秒