Usar funções de agregação com matrizes
-
Para adicionar valores dentro de uma matriz, use
SUM
, como no exemplo a seguir. -
Para agregar várias linhas dentro de uma matriz, use
array_agg
. Para ter mais informações, consulte Criar matrizes com base em subconsultas.
nota
Desde o mecanismo do Athena versão 2, ORDER BY
é permitido em funções de agregação.
WITH dataset AS ( SELECT ARRAY [ ARRAY[1,2,3,4], ARRAY[5,6,7,8], ARRAY[9,0] ] AS items ), item AS ( SELECT i AS array_items FROM dataset, UNNEST(items) AS t(i) ) SELECT array_items, sum(val) AS total FROM item, UNNEST(array_items) AS t(val) GROUP BY array_items;
Na última instrução do SELECT
, em vez de usar sum()
e UNNEST
, você pode usar reduce()
para reduzir o tempo de processamento e a transferência de dados, como no exemplo a seguir.
WITH dataset AS ( SELECT ARRAY [ ARRAY[1,2,3,4], ARRAY[5,6,7,8], ARRAY[9,0] ] AS items ), item AS ( SELECT i AS array_items FROM dataset, UNNEST(items) AS t(i) ) SELECT array_items, reduce(array_items, 0 , (s, x) -> s + x, s -> s) AS total FROM item;
As consultas retornam os seguintes resultados. A ordem de resultados obtidos não é garantida.
+----------------------+
| array_items | total |
+----------------------+
| [1, 2, 3, 4] | 10 |
| [5, 6, 7, 8] | 26 |
| [9, 0] | 9 |
+----------------------+