在 Presto 中添加两个地图

Question

我的数据如下所述

customer_id   usage_month  usage_by_product         usage
1             June         {"A":50, "B":50}         100
1             July         {"A":50, "B":10, "C":20} 80
1             Aug          {"A":50, "D":500}        550
1             Sep          {"C" :30}                30

我想编写一个汇总全年总使用量的查询

customer_id   usage_by_product    usage
 1            {"A": 150, "B":60   760
               "C": 50, "D":500}

是否可以在 Athena (Presto) 中的地图上进行这种聚合？

Answer 1

您可以使用map_entries + UNNEST 将地图拆分为单独的键/值对。然后，将值求和并聚合回map。

例如：

WITH input AS (
    SELECT * FROM (VALUES
        (1, map(array['a', 'c'], array[50, 42])), 
        (1, map(array['a', 'b'], array[50, 18]))
    ) t(customer_id, m)
),
sum_by_map_key AS (
    SELECT customer_id, k, sum(v) AS s
    FROM input
    CROSS JOIN UNNEST(map_entries(m)) AS u(k, v)
    GROUP BY customer_id, k
)
SELECT customer_id, map_agg(k, s)
FROM sum_by_map_key
GROUP BY customer_id;

输出：

 customer_id |        _col1
-------------+---------------------
           1 | {a=100, b=18, c=42}
(1 row)

注意： 要像这样添加两张地图，您可以使用map_zip_with。但是，要在聚合多行时使用它，您可能需要将所有映射值聚合到单个 array 并在其上运行数组缩减。根据这些数组的大小，将所有地图聚合到单个 array(map) 中可能有效，也可能无效。