Question

Given an array like [ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10], I want to get a random value that takes into consideration the position.

I want the likelihood of 1 popping up to be way bigger than 10.

Is something like this possible?

Answer 1

For the sake of simplicity let’s assume an array arr = [x, y, z] from which we will be sampling values. We’d like to see following relative frequencies of x, y and z:

frequencies = [5, 2, 1]

Preprocess these frequencies to calculate margins for our subsequent dice roll:

thresholds = frequencies.clone
1.upto(frequencies.count - 1).each { |i| thresholds[i] += thresholds[i - 1] }

Let’s sum them up.

max = frequencies.reduce :+

Now choose a random number

roll = 1 + rand max
index = thresholds.find_index { |x| roll <= x }

Return arr[index] as a result. To sum up:

def sample arr, frequencies
  # assert arr.count == frequencies.count
  thresholds = frequencies.clone
  1.upto(frequencies.count - 1).each { |i| thresholds[i] += thresholds[i - 1] }
  max = frequencies.reduce :+
  roll = 1 + rand(max)
  index = thresholds.find_index { |x| roll <= x }
  arr[index]
end

Let’s see how it works.

data = 80_000.times.map { sample [:x, :y, :z], [5, 2, 1] }

A histogram for data shows that sample works as we’ve intended.