10

我的数据处理中的一个常见模式是按一组列分组，应用过滤器，然后再次展平。例如：

my_data_grouped = group my_data by some_column;
my_data_grouped = filter my_data_grouped by <some expression>;
my_data = foreach my_data_grouped flatten(my_data);

这里的问题是，如果my_data在此操作之后以类似 (c1, c2, c3) 的模式开始，它将具有类似 (mydata::c1, mydata::c2, mydata::c3) 的模式。如果列是唯一的，有没有办法轻松去除“mydata::”前缀？

我知道我可以做这样的事情：

my_data = foreach my_data generate c1 as c1, c2 as c2, c3 as c3;

但是，对于具有大量列的数据集，这会变得尴尬且难以维护，而对于具有可变列的数据集来说，这是不可能的。

4

2 回答 2

4

如果架构中的所有字段都具有相同的前缀集（例如 group1::id、group1::amount 等），您可以在引用特定字段时忽略前缀（并将它们引用为 id、amount 等）

或者，如果您仍然希望去除单级前缀的架构，您可以使用这样的 UDF：

public class RemoveGroupFromTupleSchema extends EvalFunc<Tuple> {

@Override
public Tuple exec(Tuple input) throws IOException {
    Tuple result = input;
    return result;
}


@Override
public Schema outputSchema(Schema input) throws FrontendException {
    if(input.size() != 1) {
        throw new RuntimeException("Expected input (tuple) but input does not have 1 field");
    }

    List<Schema.FieldSchema> inputSchema = input.getFields();
    List<Schema.FieldSchema> outputSchema = new ArrayList<Schema.FieldSchema>(inputSchema);
    for(int i = 0; i < inputSchema.size(); i++) {
        Schema.FieldSchema thisInputFieldSchema = inputSchema.get(i);
        String inputFieldName = thisInputFieldSchema.alias;
        Byte dataType = thisInputFieldSchema.type;

        String outputFieldName;
        int findLoc = inputFieldName.indexOf("::");
        if(findLoc == -1) {
            outputFieldName = inputFieldName;
        }
        else {
            outputFieldName = inputFieldName.substring(findLoc+2);
        }
        Schema.FieldSchema thisOutputFieldSchema = new Schema.FieldSchema(outputFieldName, dataType);
        outputSchema.set(i, thisOutputFieldSchema);
    }

    return new Schema(outputSchema);
}
}
于 2012-11-26T20:11:57.340 回答
1

您可以将“AS”语句与“foreach”放在同一行。

IE

my_data_grouped = group my_data by some_column;
my_data_grouped = filter my_data_grouped by <some expression>;
my_data = FOREACH my_data_grouped FLATTEN(my_data) AS (c1, c2, c3);

但是，这与在 2 行上执行此操作相同，并且不会缓解您对“具有可变列的数据集”的问题。

于 2012-06-19T15:31:14.440 回答