Python elementwiseの例

コード例 #1

def create_add_one_struct_udf(result_formatter):
    return elementwise(
        input_type=[dt.double],
        output_type=dt.Struct(['col1', 'col2'], [dt.double, dt.double]),
    )(_format_struct_udf_return_type(add_one_struct, result_formatter))

コード例 #2

def create_add_one_udf(result_formatter):
    return elementwise(input_type=[dt.double],
                       output_type=dt.double)(_format_udf_return_type(
                           add_one, result_formatter))