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

需求

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

代码：

新建一个Random类：

<?php

namespace app\helpers;

class Random
{
    /**
     * 获取浮点随机数
     * @param int min
     * @param intmax
     * @return int
     */
    public static function randomFloat(min = 0,max = 1) {

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

    }

    /**
     * 根据概率提取数组中元素
     * @param arrayseq  待取数组
     * @param array prob 对应每个元素概率（浮点数，和等于1）
     * @param intk 重复次数
     * @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;
                }
            }
        }
        returnlist;
    }

}

他包含两个方法，一个用于获取浮点随机数，一个从指定数组$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,
)

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