Generate random number with given probability matlab

by

The simple solution is to generate a number with a uniform distribution (using rand), and manipulate it a bit:

r = rand;
prob = [0.5, 0.1, 0.4];
x = sum(r >= cumsum([0, prob]));

or in a one-liner:

x = sum(rand >= cumsum([0, 0.5, 0.1, 0.4]));

Explanation

Here r is a uniformly distributed random number between 0 and 1. To generate an integer number between 1 and 3, the trick is to divide the [0, 1] range into 3 segments, where the length of each segment is proportional to its corresponding probability. In your case, you would have:

  • Segment [0, 0.5), corresponding to number 1.
  • Segment [0.5, 0.6), corresponding to number 2.
  • Segment [0.6, 1], corresponding to number 3.

The probability of r falling within any of the segments is proportional to the probabilities you want for each number. sum(r >= cumsum([0, prob])) is just a fancy way of mapping an integer number to one of the segments.

Extension

If you’re interested in creating a vector/matrix of random numbers, you can use a loop or arrayfun:

r = rand(3); % # Any size you want
x = arrayfun(@(z)sum(z >= cumsum([0, prob])), r);

Of course, there’s also a vectorized solution, I’m just too lazy to write it.

More Related Contents:

  1. How to convert dbm to watts on Matlab
  2. Random numbers that add to 100: Matlab
  3. Generate a random point within a circle (uniformly)
  4. Select k random elements from a list whose elements have weights
  5. Weighted random numbers in MATLAB
  6. Choosing n numbers with fixed sum
  7. Generate a random number in a certain range in MATLAB
  8. Algorithm to generate Poisson and binomial random numbers?
  9. Draw random numbers from pre-specified probability mass function in Matlab
  10. How to deal with overflow and underflow?
  11. How to set custom seed for pseudo-random number generator
  12. Fitting largest circle in free area in image with distributed particle
  13. How to find out if a matrix is singular? [duplicate]
  14. How to draw a Histogram in Matlab
  15. “Undefined function ‘function_name’ for input arguments of type ‘double’.”
  16. character reconstruction and filling for OCR
  17. pdist2 equivalent in MATLAB version 7
  18. How do I resize a matrix in MATLAB?
  19. Multi-class classification in libsvm [closed]
  20. Run a script that uses multiple MATLAB sessions
  21. How to create an executable .exe file from a .m file
  22. Analytical Fourier transform vs FFT of functions in Matlab
  23. What is the closest thing MATLAB has to namespaces?
  24. Cost function in logistic regression gives NaN as a result
  25. Random points inside a parallelogram
  26. Mex files: how to return an already allocated matlab array
  27. Machine learning – Linear regression using batch gradient descent
  28. Extract arbitrarily rotated plane of data from 3D array as 2D array
  29. Select largest object in an image
  30. Efficiently compute a 3D matrix of outer products – MATLAB

Leave a Comment