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

How to create UDF in pig for categorize columns with respect to another filed

我想在 pig 中使用 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

请推荐

您可以在没有 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;

您可以在 eclipse 中创建 pig udfs

在 eclipse 中用 pig jars 创建一个项目并尝试下面的代码

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>

用包定义它 class。

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);