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