web-dev-qa-db-ja.com

確率のあるランダムな整数を生成する

確率のある整数値を生成する方法について少し混乱しています。

例として、確率値を持つ4つの整数があります:1 | 0.4、2 | 0.3、3 | 0.2、4 | 0.1

確率を考慮してこれらの4つの数値を生成するにはどうすればよいですか?

23
Headshota

ここに便利なトリックがあります:-)

function randomWithProbability() {
  var notRandomNumbers = [1, 1, 1, 1, 2, 2, 2, 3, 3, 4];
  var idx = Math.floor(Math.random() * notRandomNumbers.length);
  return notRandomNumbers[idx];
}
36

単純な素朴なアプローチは次のとおりです。

function getRandom(){
  var num=Math.random();
  if(num < 0.3) return 1;  //probability 0.3
  else if(num < 0.6) return 2; // probability 0.3
  else if(num < 0.9) return 3; //probability 0.3
  else return 4;  //probability 0.1
}
28
bhups

@bhupsの回答に基づくより柔軟なソリューション。これは、確率値(重み)の配列を使用します。 'weights'要素の合計は1に等しくなければなりません。

var weights = [0.3, 0.3, 0.3, 0.1]; // probabilities
var results = [1, 2, 3, 4]; // values to return

function getRandom () {
    var num = Math.random(),
        s = 0,
        lastIndex = weights.length - 1;

    for (var i = 0; i < lastIndex; ++i) {
        s += weights[i];
        if (num < s) {
            return results[i];
        }
    }

    return results[lastIndex];
};
14
Rom098

確率と残りの乱数を継続的にチェックすることをお勧めします。

この関数は、最初に戻り値を可能な最後のインデックスに設定し、残りのランダム値が実際の確率より小さくなるまで繰り返します。

確率の合計は1になる必要があります。

function getRandomIndexByProbability(probabilities) {
    var r = Math.random(),
        index = probabilities.length - 1;

    probabilities.some(function (probability, i) {
        if (r < probability) {
            index = i;
            return true;
        }
        r -= probability;
    });
    return index;
}

var i,
    probabilities = [0.4, 0.3, 0.2, 0.09, 0.01 ],
    count = {},
    index;

probabilities.forEach(function (a) { count[a] = 0; });

for (i = 0; i < 1e6; i++) {
    index = getRandomIndexByProbability(probabilities);
    count[probabilities[index]]++
}

console.log(count);
4
Nina Scholz

これは、確率のあるオブジェクトのセット内で選択するための、私が最も柔軟だと思うソリューションです。

// set of object with probabilities:
const set = {1:0.4,2:0.3,3:0.2,4:0.1};

// get probabilities sum:
var sum = 0;
for(let j in set){
    sum += set[j];
}

// choose random integers:
console.log(pick_random());

function pick_random(){
    var pick = Math.random()*sum;
    for(let j in set){
        pick -= set[j];
        if(pick <= 0){
            return j;
        }
    }
}
1