从具有加权行概率的PostgreSQL表中选择随机行

前端之家收集整理的这篇文章主要介绍了从具有加权行概率的PostgreSQL表中选择随机行前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
示例输入:
SELECT * FROM test;
 id | percent   
----+----------
  1 | 50 
  2 | 35   
  3 | 15   
(3 rows)

你如何编写这样的查询,平均50%的时间我可以得到id = 1的行,35%的时间行id = 2,15%的时间行id = 3?

我尝试像SELECT id FROM测试ORDER BY p * random()DESC LIMIT 1,但它给出错误的结果. 10,000次运行后,我得到一个分布,如:{1 = 6293,2 = 3302,3 = 405},但我预计分布接近:{1 = 5000,2 = 3500,3 = 1500}.

有任何想法吗?

这应该是诀窍:
WITH CTE AS (
    SELECT random() * (SELECT SUM(percent) FROM YOUR_TABLE) R
)
SELECT *
FROM (
    SELECT id,SUM(percent) OVER (ORDER BY id) S,R
    FROM YOUR_TABLE CROSS JOIN CTE
) Q
WHERE S >= R
ORDER BY id
LIMIT 1;

查询Q给出以下结果:

1  50
2  85
3  100

然后我们在范围[0,100]中简单地生成一个随机数,并选择该数字以上的第一行(WHERE子句).我们使用通用表表达式(WITH)来确保随机数仅计算一次.

BTW,SELECT SUM(percent)FROM YOUR_TABLE允许您以百分比表示任何权重 – 它们不会严格需要百分比(即加起来为100).

[SQL Fiddle]

原文链接:https://www.f2er.com/postgresql/192719.html

猜你在找的Postgre SQL相关文章