为此,您需要使用带有嵌套 LIMIT 的 GROUP BY、FOREACH,而不是 JOIN 或 COGROUP。请参阅 Pig 0.10 中的实现,我使用您输入的数据来获得指定的输出:
A = load '~/pig/data/subset_join_A.dat' as (k:chararray, m:chararray, n:chararray);
B = load '~/pig/data/subset_join_B.dat' as (o:chararray, p:int);
-- as join will be on m, we need to leave only 2 rows per a value in m.
group_A = group A by m;
top_A_x = foreach group_A {
top = limit A 2; -- where x = 2
generate flatten(top);
};
-- another way to do join, allows us to do left or right joins and checks
co_join = cogroup top_A_x by (m), B by (o);
-- filter out records from A that are not in B
filter_join = filter co_join by IsEmpty(B) == false;
result = foreach filter_join generate flatten(top_A_x);
或者,您可以仅使用 COGROUP、FOREACH 和嵌套 LIMIT 来实现它:
A = load '~/pig/data/subset_join_A.dat' as (k:chararray, m:chararray, n:chararray);
B = load '~/pig/data/subset_join_B.dat' as (o:chararray, p:int);
co_join = cogroup A by (m), B by (o);
filter_join = filter co_join by IsEmpty(B) == false;
result = foreach filter_join {
top = limit A 2;
--you can limit B as well
generate flatten(top);
};