如何在pig中创建UDF,以便根据另一个字段对列进行分类

时间:2017-10-17 16:57:04

标签: hadoop hdfs apache-pig pig-udf

我想在猪中使用UDF对一列相对于其他列进行分类。

我有数据

Id,name,age
1,jhon,31
2,adi,15
3,sam,25
4,lina,28

预期输出

1,jhon,31,30-35
2,adi,15,10-15
3,sam,25,20-25
4,lina,28,25-30

请建议

2 个答案:

答案 0 :(得分:1)

您可以在没有UDF的情况下执行此操作。假设您已将数据加载到关系A。

B = FOREACH A GENERATE A.Id,A.name,A.age,(A.age%5 == 0 ? A.age-5 : (A.age/5)*5) as lower_age,(A.age%5 == 0 ? A.age : ((A.age/5)*5) + 5) as upper_age;
C = FOREACH B GENERATE B.Id,B.name,B.age,CONCAT(CONCAT((chararray)lower_age,'-'),(chararray)upper_age);
DUMP C;

答案 1 :(得分:0)

你可以在eclipse中创建pig udfs

使用猪罐子在eclipse中创建一个项目并尝试下面的代码

package com;

import java.io.IOException;

import org.apache.pig.EvalFunc;
import org.apache.pig.backend.executionengine.ExecException;
import org.apache.pig.data.Tuple;



public class Age extends EvalFunc<String>{

    @Override
    public String exec(Tuple a) throws IOException {
        // TODO Auto-generated method stub
        if(a == null || a.size() == 0){
            return null;
        }
        try{
            Object object = a.get(0);
            if(object == null){
                return null;
            }
            int i = (Integer) object;
            if(i >= 10 && i <= 20 ){
                return "10-20";
            }
            else if (i >= 21 && i <= 30){
                return "20-30";
            }
            else 
                return ">30";
        } catch (ExecException e){
            throw new IOException(e);
        }
    }

}

现在将项目导出为jar并将其注册到pig shell

REGISTER <path of your .jar file>

使用包和类定义它。

DEFINE U com.Age();

a = LOAD '<input path>' using PigStorage(',') as (id:int,name:chararray,age:int);

b = FOREACH a GENERATE id,name,age,U(age);