如何将列表传递给pyspark中的UserDefinedFunction（UDF）

Question

我需要传递一个列表作为pyspark中某个UDF的参数。 例：

def cat(mine,mine2):
     if mine is not None and mine2 is not None:
             return "2_"+mine+"_"+mine2

udf_cat = UserDefinedFunction(cat, "string")

l = ["COLUMN1","COLUMN2"]

df = df.withColumn("NEW_COLUMN", udf_cat(l))

但是我总是会出错。

Answer 1

过了一会儿，我发现我所需要的只是在列表前使用字符“ *”传递。 例：

df = df.withColumn("NEW_COLUMN", udf_cat(*l))

这样，它将起作用。