arrays - （轮盘）对数组中的多个元素进行排序

Question

我正在执行轮盘赌选择（http://www.edc.ncl.ac.uk/assets/hilite_graphics/rhjan07g02.png），我想计算n元素的选择。我怎样才能避免使用循环？

例如，我有以下prob向量：

prob = [0.1 0.3 0.4 0.15 0.05];

单曲的选择element=0.2是：

cumprob = cumsum(prob);
selected = find(element<=prob,1,'first')

选择 = 2

但是，计算n元素的选择呢？直观而缓慢的方法是：

cumprob = cumsum(prob);
for id = 1:1:n 
   selected(id) = find(element(id)<=prob,1,'first');
end

有什么方法可以避免使用 for 循环来实现这一点？

提前致谢。

Answer 1

方法一：使用discretize（需要Matlab版本R2015a或更高版本）

方法2：使用arrayfun速度较慢的方法。

测试代码：

n = 5e6;
element = rand(n,1);
prob = [0.1 0.3 0.4 0.15 0.05];

cumprob = cumsum(prob);

tic
selected1 = zeros(n,1);
for id = 1:1:n 
   selected1(id) = find(element(id)<=cumprob,1,'first');
end
toc

tic
selected2 = discretize(element,[0,cumprob]);
toc
isequal(selected1, selected2)

tic
selected3 = arrayfun(@(e) sum(e>=cumprob)+1, element);
toc
isequal(selected1, selected3)

时序（不准确，但有效）和准确度比较：

Elapsed time is 5.634721 seconds.
Elapsed time is 0.059813 seconds.

ans =

     1

Elapsed time is 18.838859 seconds.

ans =

     1

>>