php 输出任意一个数,PHP按指定概率随机获取数组元素

需求

客户跟进项目，当有新客户进入系统，按照指定概率将客户分配给指定员工来跟进，给每个待指定的员工一个概率(概率之和为1)，然后按照概率来分配即可。

代码：

新建一个Random类：

namespace app\helpers;

class Random

{

/**

* 获取浮点随机数

* @param int $min

* @param int $max

* @return int

*/

public static function randomFloat($min = 0, $max = 1) {

return $min + mt_rand() / mt_getrandmax() * ($max - $min);

}

/**

* 根据概率提取数组中元素

* @param array $seq 待取数组

* @param array $prob 对应每个元素概率(浮点数，和等于1)

* @param int $k 重复次数

* @return array

*/

public static function Choose($seq, $prob, $k=1)

{

$list = [];

for ($i = 0; $i < $k; $i++) {

$q = self::randomFloat();

for ($j = 0; $j <= count($seq); $j++) {

if(array_sum(array_slice($prob, 0, $j)) < $q && $q <= array_sum(array_slice($prob, 0, $j + 1))) {

$list[$i] = $seq[$j];

break;

}

}

}

return $list;

}

}

他包含两个方法，一个用于获取浮点随机数，一个从指定数组$seq，按指定概率$prob获取$k次元素。

运行测试

$a = ['a', 'b', 'c', 'd'];

$b = [0.25, 0.50, 0.25, 0];

// 重复100000次

$k = 100000;

$res = Random::Choose($a, $b, $k);

// 统计每个元素出现的次数

var_export(array_count_values($res));

输出

array (

'b' => 50140,

'c' => 24791,

'a' => 25069,

)

多次运行会发现，已经实现了按照指定概率来提取元素，当然，次数越多越接近。

PHP按指定概率随机获取数组元素

更多精彩，敬请关注本博微信公众号：hsu1943