如何对数组进行排序并返回配置单元中的索引？

Posted 2023-03-23

【中文标题】如何对数组进行排序并返回配置单元中的索引？

【英文标题】：How to sort an array and return the index in hive?

【发布时间】：2020-05-21 13:02:01

【问题描述】：

在hive中，我希望将一个数组从大到小排序，得到索引数组。

比如表格是这样的：

id  |  value_array
 1  |  30, 40, 10, 20
 2  |  10, 30, 40, 20

我想得到这个：

id  |  value_array
 1  |  1, 0, 3, 2
 2  |  2, 1, 3, 0

结果中的数组是初始元素的索引。我怎样才能做到这一点？

【参考方案1】：

使用poseexplode分解数组得到索引和值，按值排序，收集索引数组：

select id, collect_list(pos) as result_array
from
(
select s.id, a.pos, a.v 
  from your_table s
       lateral view posexplode(s.value_array) a as pos, v
distribute by s.id sort by a.v DESC --sort by value
)s
group by id
;

测试，结果：

id  result_array
1   [1,0,3,2]
2   [2,1,3,0]

【讨论】：

@VamsiPrabhala "从大到小" 2->40, 1->30, 以此类推

@VamsiPrabhala 好的，np