Skip to content

Instantly share code, notes, and snippets.

@sllynn
Created June 15, 2020 15:48
Show Gist options
  • Save sllynn/bd013753716a223bb666eea88588257a to your computer and use it in GitHub Desktop.
Save sllynn/bd013753716a223bb666eea88588257a to your computer and use it in GitHub Desktop.
Convert VectorUDT to ArrayType (Scala UDF)
import org.apache.spark.ml.linalg.Vector
val toArray = udf { v: Vector => v.toArray }
spark.sqlContext.udf.register("toArray", toArray)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment