hive - 如何在 Hive 中获取数组中的前 n 个元素

Question

我在Hive中使用split函数创建一个数组，如何从数组中获取前n个元素，我想遍历子数组

代码示例

select col1 from table
where split(col2, ',')[0:5] 

'[0:5]'看起来像python风格，但在这里不起作用。

Answer 1

这是一种更简单的方法。有一个UDF here称为 TruncateArrayUDF.java 可以执行您所要求的操作。只需克隆 repo从主页并使用 Maven 构建 jar .

示例数据:

|       col1         |
----------------------
  1,2,3,4,5,6,7
  11,12,13,14,15,16,17

查询:

add jar /complete/path/to/jar/brickhouse-0.7.0-SNAPSHOT.jar;
create temporary function trunc as 'brickhouse.udf.collect.TruncateArrayUDF';

select pos
      ,newcol
from (
      select trunc(split(col1, '\\,'), 5) as p
      from table
     ) x
lateral view posexplode(p) explodetable as pos, newcol

输出:

  pos  |  newcol  |
-------------------
  0         1
  1         2
  2         3
  3         4
  4         5
  0         11
  1         12
  2         13
  3         14
  4         15