为每个组选择随机行

时间:2013-04-16 18:52:21

标签: sql postgresql random group-by window-functions

我有一张这样的表

ID    ATTRIBUTE
 1    A
 1    A
 1    B
 1    C
 2    B
 2    C
 2    C
 3    A
 3    B
 3    C

我想为每个ID选择一个随机属性。因此结果看起来像这样(虽然这只是众多选项中的一个

ATTRIBUTE
B
C
C

这是我对此问题的尝试

SELECT
  "ATTRIBUTE"
FROM
  (
  SELECT
    "ID",
    "ATTRIBUTE",
    row_number() OVER (PARTITION BY "ID" ORDER BY random()) rownum
  FROM
    table
  ) shuffled
WHERE
  rownum = 1

然而,我不知道这是否是一个很好的解决方案,因为我需要引入行号,这有点麻烦。

你有更好的吗?

2 个答案:

答案 0 :(得分:15)

select distinct on (id) id, attribute
from like_this
order by id, random()

如果您只需要属性列:

select distinct on (id) attribute
from like_this
order by id, random()

请注意,您仍然需要先id订购,因为它是distinct on的一列。

如果您只想要不同的属性:

select distinct attribute
from (
    select distinct on (id) attribute
    from like_this
    order by id, random()
) s

答案 1 :(得分:0)

在每条记录(id)前放一个大的随机数,并在每组中选择随机数最低的记录。

$ cat test.txt
\N  1   a
\N  2   b
\N  2   c
\N  2   d
\N  3   e
\N  4   f


$ mysql

USE test;
DROP TABLE test;
CREATE TABLE test (id0 INT NOT NULL AUTO_INCREMENT, id VARCHAR(1),  attribute VARCHAR(1), PRIMARY KEY (id0));
LOAD DATA LOCAL INFILE '~/mysql/test.txt' INTO TABLE test FIELDS TERMINATED BY '\t';

DROP TABLE rtest;
CREATE TABLE rtest (random INT(8), id0 VARCHAR(1), id VARCHAR(1),  attribute VARCHAR(1),  PRIMARY KEY (id, random));

INSERT INTO rtest
SELECT CAST(1000000. * rand() AS INT) AS random, test.* FROM test;

SELECT rtest.* FROM rtest,
(SELECT id, min(random) AS random FROM rtest GROUP BY id) AS sample WHERE rtest.random=sample.random AND rtest.id=sample.id;