如何將列表傳遞給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))

這樣，它將起作用。