如何在猪中使用CASE语句？

Question

我需要以下输出。

NE 50
SE 80

我正在使用猪查询来计算基于区域的国家/地区。

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 没有 case 语句。最好的选择是使用 UDF。如果组值仅限于两个，那么您可以使用bincond运算符来检查值

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

如果您有多个值，则使用 this.Ive 使用测试值来生成输出。

输入

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

输出