import pyspark.sql.types as T @udf(returnType=T.ArrayType(T.ArrayType(T.DoubleType()))) def aad(): return [[1.22]] df.withColumn("new_col", aad(F.col()))
import pyspark.sql.types as T @udf(returnType=T.ArrayType(T.ArrayType(T.DoubleType()))) def aad(): return [[1.22]] df.withColumn("new_col", aad(F.col()))