Presto的find_first_index数组函数在AWS Athena中的等效替代方案

Presto的find_first_index函数用于返回数组中首个满足指定条件的元素的0-based索引，无匹配时返回NULL。在AWS Athena中可以通过以下两种方案实现相同逻辑：

方案一：使用UNNEST WITH ORDINALITY展开数组 #

这种方法适合处理复杂的筛选条件（比如涉及表中其他列的计算），步骤是展开数组并生成元素的1-based索引，再筛选出符合条件的最小索引，最后转换为0-based结果：

SELECT
  t.*,
  -- 转换为0-based索引，若不需要则直接使用matched.min_ordinal
  CASE WHEN matched.min_ordinal IS NOT NULL THEN matched.min_ordinal - 1 ELSE NULL END AS first_matching_index
FROM your_table t
LEFT JOIN (
  SELECT
    -- 替换为你表中唯一标识每行的字段（如主键id）
    unique_row_id,
    MIN(ordinal) AS min_ordinal
  FROM your_table
  CROSS JOIN UNNEST(target_array) WITH ORDINALITY AS elem(val, ordinal)
  -- 替换为你的筛选条件，例如val > 10或val LIKE '%test%'
  WHERE elem.val > 10
  GROUP BY unique_row_id
) matched ON t.unique_row_id = matched.unique_row_id;

方案二：使用filter+array_position函数组合 #

这种方法更简洁，适合简单的筛选逻辑，利用filter提取所有符合条件的元素，再取第一个元素的位置：

SELECT
  target_array,
  -- 转换为0-based索引，若需要1-based则去掉-1
  CASE WHEN filtered_arr[1] IS NOT NULL THEN array_position(target_array, filtered_arr[1]) - 1 ELSE NULL END AS first_matching_index
FROM (
  SELECT
    target_array,
    -- 替换为你的筛选逻辑，例如x -> x < 5或x = 'target_value'
    filter(target_array, x -> x < 5) AS filtered_arr
  FROM your_table
) subquery;

注意事项 #

• Athena的数组索引默认是1-based，因此需要通过-1转换为与find_first_index一致的0-based结果；
• 方案二中如果数组存在多个相同的符合条件元素，array_position会返回第一个出现的位置，与原函数逻辑一致；
• 若筛选条件涉及复杂计算（比如关联其他列的值），优先选择方案一。

内容的提问来源于stack exchange，提问作者Roelant