Sqlite / 填充对现有行进行排名的新列答案

【问题标题】：Sqlite / populate new column that ranks the existing rowsSqlite / 填充对现有行进行排名的新列
【发布时间】：2021-10-27 21:26:15
【问题描述】：

我有一个包含以下列的 SQLite 数据库表：

| day         | place | visitors |
-------------------------------------
|  2021-05-01 | AAA   |   20 |
|  2021-05-01 | BBB   |   10 |
|  2021-05-01 | CCC   |    3 |
|  2021-05-02 | AAA   |    5 |
|  2021-05-02 | BBB   |    7 |
|  2021-05-02 | CCC   |    2 |

现在我想介绍一个列“排名”，它表示每天访问者的排名。预期的表格如下所示：

| day         | place | visitors | Rank  |
------------------------------------------
|  2021-05-01 | AAA   |   20     |  1    |
|  2021-05-01 | BBB   |   10     |  2    |
|  2021-05-01 | CCC   |    3     |  3    |
|  2021-05-02 | AAA   |    5     |  2    |
|  2021-05-02 | BBB   |    7     |  1    |
|  2021-05-02 | CCC   |    2     |  3    |

可以使用（伪代码）之类的程序来填充新列 Rank 的数据。

for each i_day in all_days:
    SELECT
    ROW_NUMBER () OVER (ORDER BY `visitors` DESC) Day_Rank, place
    FROM mytable
    WHERE `day` = 'i_day'
    
    for each i_place in all_places:
        UPDATE mytable 
        SET rank= Day_Rank
        WHERE `Day`='i_day'
        AND place = 'i_place'

由于这种逐行更新效率很低，我正在寻找如何结合 UPDATE 使用 SQL 子查询来优化它。

（目前还不行……）

for each i_day in all_days:
    UPDATE mytable
    SET rank= (
    SELECT
        ROW_NUMBER () OVER (ORDER BY `visitors` DESC) Day_Rank
        FROM mytable
        WHERE `day` = 'i_day'
        )

【问题讨论】：

请同时包含预期的输出表。

标签： sqlite sql-update subquery window-functions rank

【解决方案1】：

通常，这可以通过子查询来完成，该子查询计算visitors 大于当前行的visitors 值的行数：

UPDATE mytable
SET Day_Rank = (
  SELECT COUNT(*) + 1
  FROM mytable m 
  WHERE m.day = mytable.day AND m.visitors > mytable.visitors 
);

请注意，如果visitors 的值存在关联，则结果实际上是RANK() 将返回的结果。

请参阅demo。

或者，您可以在 CTE 中使用 ROW_NUMBER() 计算排名并在子查询中使用它：

WITH cte AS (
  SELECT *, ROW_NUMBER() OVER (PARTITION BY day ORDER BY visitors DESC) rn
  FROM mytable
)
UPDATE mytable
SET Day_Rank = (SELECT rn FROM cte c WHERE (c.day, c.place) = (mytable.day, mytable.place));

请参阅demo。

或者，如果您的 SQLite 版本是 3.33.0+，您可以使用类似连接的 UPDATE...FROM... 语法：

UPDATE mytable AS m
SET Day_Rank = t.rn
FROM (
  SELECT *, ROW_NUMBER() OVER (PARTITION BY day ORDER BY visitors DESC) rn
  FROM mytable
) t
WHERE (t.day, t.place) = (m.day, m.place);

【讨论】：