Question

Change the following code to give only the first 2 picks; or 3 picks (Matlab)

% colors for ball

colors = ["red", "blue", "green", "yellow"];

% initial probabilites

% change as required

p = [0.20, 0.25, 0.35, 0.20];

% cumulative sum of p

c = cumsum(p);

% continue drawing until all are drawn

while(size(p) > 0)

% generate a random number

r = rand();

% draw ball according to random number

ball = 0;

for i = 1:size(p, 2)

if(r <= c(i))

% draw ith ball

ball = i;

break;

end

end

fprintf("%s ball is drawn.\n", colors(ball))

% delete ball

p(ball) = [];

colors(ball) = [];

% redistribute probabilities

p = p ./ sum(p);

c = cumsum(p);

end

Answer 1

colors = ["red", "blue", "green", "yellow"];

p = [0.20, 0.25, 0.35, 0.20];

c = cumsum(p);

while(size(p) > 0)

r = rand();

ball = 0;

for i = 1:size(p, 2)

if(r <= c(i))

% draw ith ball

ball = i;

break;

end

end

fprintf("%s ball is drawn.\n", colors(ball))

% delete ball

p(ball) = [];

colors(ball) = [];

p = p ./ sum(p);

c = cumsum(p);

end