hadoop - 如何在 pig 中使用 CASE 语句？

Question

我需要以下输出。

NE 50
SE 80

我正在使用 pig 查询来根据地区统计国家/地区。

c1 = group country by zone;
c2 = foreach c1 generate COUNT(country.zone), (
case country.zone
when 1 then 'NE'
else 'SE'
);

但我无法实现我的输出。我收到如下错误:

2016-03-30 13:57:16,569 [main] ERROR org.apache.pig.tools.grunt.Grunt - ERROR 1039: (Name: Equal Type: null Uid: null)incompatible types in Equal Operator left hand side:bag :tuple(zone:int)  right hand side:int
Details at logfile: /home/cloudera/pig_1459370643493.log

但我能够使用以下查询。

c2 = foreach c1 generate group, COUNT(country.zone);

这将给出以下输出:

(1,50)
(2,80)

如何添加 NE 而不是 1 和 SE 而不是 2？我认为使用 CASE 会有所帮助，但我遇到了错误。谁能帮忙？

Answer 1

编辑

Pig 0.12.0版本支持CASE表达式。

c2 = FOREACH c1 GENERATE (CASE group
                              WHEN 1 THEN 'NE' 
                              WHEN 2 THEN 'SE'
                              WHEN 3 THEN 'AE'
                              ELSE 'VR' END), COUNT(country.zone);

旧 pig 版本

Pig 没有 case 语句。你最好的选择是使用 UDF。如果组值仅限于两个，那么你可以使用 bincond运算符检查值

c2 = foreach c1 generate (group == 1 ? 'NE' : 'SE'), COUNT(country.zone);

如果你有多个值，那么使用这个。我使用测试值来生成输出。

输入

c2 = FOREACH c1 GENERATE (group == 1 ? 'NE' : 
                         (group == 2 ? 'SE' :
                         (group == 3 ? 'AE' : 'VR'))), COUNT(country.zone);

输出