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

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

发表评论

电子邮件地址不会被公开。 必填项已用*标注