主页 / dba / 问题 / 55363
Accepted
korda
Asked: 2013-12-21 03:29:46 +0800 CST

从集合中设置随机值

  • 772

我需要将一些随机值放入数据库，但我不想最终得到完全随机的文本（如 7hfg43d3）。相反，我想随机选择我自己提供的值之一。

postgresql random

3 个回答

  1. Best Answer

    好主意。我建议做两个小的简化：

    ('{Foo,Bar,Poo}'::text[])[ceil(random()*3)]

    • 使用数组字面量( '{Foo,Bar,Poo}'::text[]) 的更简单语法 为更长的列表缩短字符串。额外的好处：显式类型声明适用于任何类型，而不仅仅是text. 您最初的想法恰好是 output text，因为这是字符串文字的默认类型。

    • 使用ceil()而不是floor() + 1. 结果相同。

    好的，从理论上讲，正如您的评论中所暗示的那样，下限可以精确地为 0 ，因为random()产生（在此处引用手册）：

    0.0 <= x < 1.0 范围内的随机值

    但是，我从未见过这种情况发生。运行几百万次测试：

    SELECT count(*)
FROM   generate_series(1,1000000)
WHERE  ceil(random())::int = 0;

    -> SQL小提琴

    不过，为了绝对安全，您可以使用 Postgres 自定义数组下标，并且仍然避免额外添加：

    ('[0:2]={Foo,Bar,Poo}'::text[])[floor(random()*3)]

    看：

    更好的是，使用trunc()，这有点快。

    ('[0:2]={Foo,Bar,Poo}'::text[])[trunc(random()*3)]
    • 31

  2. 我想出了使用数组来完成此任务的想法：

    (ARRAY['Foo','Bar','Poo'])[floor(random()*3)+1]
    • 9

  3. 基于这个想法，我创建了一个对我非常有用的函数：

    CREATE OR REPLACE FUNCTION random_choice(
    choices text[]
)
RETURNS text AS $$
DECLARE
    size_ int;
BEGIN
    size_ = array_length(choices, 1);
    RETURN (choices)[floor(random()*size_)+1];
END
$$ LANGUAGE plpgsql;

    使用示例：

    • SELECT random_choice(array['h', 'i', 'j', 'k', 'l']) as random_char;

    • SELECT random_choice((SELECT array_agg(name) FROM pets)) AS pet_name;

    • 1

相关问题