python - 是否可以将标量参数传递给 Python UDF？

Question

是否可以按如下方式向 Python UDF 传递标量参数:

A = LOAD 'date_list' AS (date:chararray); --Example date: 20130131
B = FOREACH A GENERATE DATE_LIB.get_week('YYYYMMDD', date)

我希望能够提供一个标量参数来指定日期的格式(例如，“DD-MON-YY”代表像“31-JAN-13”这样的日期——我可以重复使用相同的日期我在 DATE_LIB 中使用控制流处理这种情况的 UDF 代码)。我将如何在 Python 中执行此操作？由于 Python UDF 只能是 EvalFunc，它们有这种能力吗？

Answer 1

这是可能的 - 如果您查看 example scripts你会看到

@outputSchemaFunction("squareSchema")
def square(num):
  return ((num)*(num))

和调用示例

b = foreach a generate myfuncs.square(3);

在你的情况下，udf 会是这样的

@outputSchemaFunction("dow:long")
def get_week(format,source_date):
    your code here