如何编写SQL查询,为特定条件选择不同的对值?

时间:2016-11-01 17:17:27

标签: sql postgresql group-by data-partitioning

我在解决以下问题时遇到问题:

对于具有特定分数的配对值,如何将它们分组,只返回具有最佳分数的不同配对值?

例如,假设我有一个包含以下行值的表:

(t1,p1,65)
(t1,p2,60)
(t1,p3,20)
(t2,p1,60)
(t2,p2,59)
(t2,p3,15)

前两列表示配对值,第三列表示配对分数。最高分为(t1,p1,65)。由于现在使用了t1和p1,我想将它们排除在进一步分析之外。

下一个最高分是(t2,p2,59)。尽管(t1,p2)得分为60,但我想排除它,因为已经使用了“t1”。 (t2,p1)也得分为60,但由于p1也已被使用,因此不包括此对。

这导致不同的配对得分值:

(t1,p1,65)
(t2,p2,59)

有没有办法只用查询生成这个结果?我试图想办法对结果进行分组和分区,但由于必须根据得分等级对已使用的值进行一些计算,我发现这很难处理。

编辑:

生成数据:

with t(t, p, score) as (
    (values ('t1','p1',65),
           ('t1','p2',60),
           ('t1','p3',20),
           ('t2','p1',60),
           ('t2','p2',59),
           ('t2','p3',15)
     ))
select t.* from t;

4 个答案:

答案 0 :(得分:4)

这个问题显然一直困扰着我。以下似乎实现了您的逻辑,将访问值的数组保存在行中:

with recursive t(t, p, score) as (
    (values ('t1','p1',65),
           ('t1','p2',60),
           ('t1','p3',20),
           ('t2','p1',60),
           ('t2','p2',59),
           ('t2','p3',15)
     )),
     cte(t, p, score, cnt, lastt, lastp, ts, ps) as (
        (select t.*, count(*) over ()::int, tt.t, tt.p, ARRAY[tt.t], ARRAY[tt.p]
         from t cross join
              (select t.* from t order by score desc limit 1) tt
        ) 
        union all
        select t, p, score, 
               sum(case when not (ts @> ARRAY[t] or ps @> ARRAY[p]) then 1 else 0 end) over ()::int,
               first_value(t) over (order by case when not (ts @> ARRAY[t] or ps @> ARRAY[p]) then score end desc nulls last),
               first_value(p) over (order by case when not (ts @> ARRAY[t] or ps @> ARRAY[p]) then score end desc nulls last),
               ts || first_value(t) over (order by case when not (ts @> ARRAY[t] or ps @> ARRAY[p]) then score end desc nulls last),
               ps || first_value(p) over (order by case when not (ts @> ARRAY[t] or ps @> ARRAY[p]) then score end desc nulls last)
        from cte 
        where cnt > 0
       )
 select *
 from cte
 where lastt = t and lastp = p and cnt > 0;

答案 1 :(得分:3)

使用存储函数相对简单:

--drop function if exists f();
--drop table if exists t;
create table t(x text,y text, z int);
insert into t values
  ('t1','p1',65),
  ('t1','p2',60),
  ('t1','p3',20),
  ('t2','p1',60),
  ('t2','p2',59),
  ('t2','p3',15)/*,
  ('t3','p1',20),
  ('t3','p2',60),
  ('t3','p3',40)*/;

create function f() returns setof t immutable language plpgsql as $$
declare
  ax text[];
  ay text[];
  r t;
begin
  ax := '{}'; ay := '{}';
  loop
    select * into r
      from t
      where x <> all(ax) and y <> all(ay)
      order by z desc, x, y limit 1;
    exit when not found;
    ax := ax || r.x; ay := ay || r.y;
    return next r;
  end loop;
end $$;

select * from f();
╔════╤════╤════╗
║ x  │ y  │ z  ║
╠════╪════╪════╣
║ t1 │ p1 │ 65 ║
║ t2 │ p2 │ 59 ║
╚════╧════╧════╝

但是,如果取消注释第三组值,结果将会有所不同:

╔════╤════╤════╗
║ x  │ y  │ z  ║
╠════╪════╪════╣
║ t1 │ p1 │ 65 ║
║ t3 │ p2 │ 60 ║
║ t2 │ p3 │ 15 ║
╚════╧════╧════╝

Upd:以及在相同测试数据上使用递归CTE的等价物:

with recursive r as (
  (select x, y, z, array[x] as ax, array[y] as ay from t order by z desc, x, y limit 1)
  union all
  (select t.x, t.y, t.z, r.ax || t.x, r.ay || t.y from t, r 
  where not (t.x = any(r.ax) or t.y = any(r.ay)) 
  order by t.z desc, t.x, t.y limit 1))
select * from r;

答案 2 :(得分:2)

使用了 t1,因此您排除了(t1,p2),但也使用了p1,但您还没有排除它。对我来说,它看起来就像是第一列的灌浆。

select t1.c1, t2.c2, t1.s 
  from table1 t2 
  inner join (select c1, max(score) s from table1 group by t1) t1 
    on (t1.s=t2.score and t1.c1=t2.c1);

其中table1是您的表的名称,c1是第一个,c2秒和score第三列;

答案 3 :(得分:1)

如果第一对值和第二对值是不同的列(例如,X和Y),您可以按X分组并执行MAX(得分)作为聚合函数,以获得以X开头的元组的最大分数。

进一步的步骤取决于您的数据,因为如果每个元组都被反转,您可能仍会得到不需要的重复项。因此,要排除这些反转元组,您可以先进行自我加入。