從數(shù)組中獲取一個隨機元素，使用提供的 weights 作為每個元素的概率

使用 Array.prototype.reduce() 方法為 weights中的每個值創(chuàng)建一個數(shù)組。

使用 Math.random() 方法生成隨機數(shù)并用 Array.prototype.findIndex() 方法根據(jù)先前生成的數(shù)組找到正確的索引。

最后，返回帶有生成索引的 arr 元素。

JavaScript

const weightedSample = (arr, weights) => {

?let roll = Math.random();

?return arr[

? ?weights

? ? ?.reduce(

? ? ? ?(acc, w, i) => (i === 0 ? [w] : [...acc, acc[acc.length - 1] + w]),

? ? ? ?[]

? ? ?)

? ? ?.findIndex((v, i, s) => roll >= (i === 0 ? 0 : s[i - 1]) && roll < v)

?];

};

示例：

weightedSample([3, 7, 9, 11], [0.1, 0.2, 0.6, 0.1]); // 9

更多內(nèi)容請訪問我的網(wǎng)站：https://www.icoderoad.com