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

需求

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

代码：

新建一个 Random 类：

<?php

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,
)

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