Athena 查詢獲取列表中任何元素的索引

Question

我需要根據另一個類似列表的列中其他元素的位置來訪問類型為列表的列中的元素。 說，我的數據集是這樣的：

WITH dataset AS (
  SELECT ARRAY ['hello', 'amazon', 'athena'] AS words,
  ARRAY ['john', 'tom', 'dave'] AS names
)
SELECT * FROM dataset

我要實現

SELECT element_at(words, index(names, 'john')) AS john_word
FROM dataset

有沒有辦法在 Athena 中擁有像“索引”這樣的功能？ 或者我怎樣才能像這樣定制一個？ 期望的結果應該是這樣的：

| -------- |
| john_word|
| -------- |
| hello    |
| -------- |

Answer 1

array_position ：

array_position(x, element) → bigint
返回數組x中元素第一次出現的位置（如果未找到，則返回 0）。

請注意，presto 數組索引從 1 開始。

SELECT element_at(words, array_position(names, 'john')) AS john_word
FROM dataset