pyspark.ml.functions.array_to_vector

pyspark.ml.functions。 array_to_vector ( 上校:pyspark.sql.column.Column )→pyspark.sql.column.Column

將數組的列數值類型轉換成一個列pyspark.ml.linalg。DenseVector實例

參數
上校 pyspark.sql.Column或str

輸入列

返回
pyspark.sql.Column

密集的轉換列向量。

例子

> > >pyspark.ml.functions進口array_to_vector> > >df1=火花createDataFrame((((1.5,2.5),),,模式=“v1陣列雙> <”)> > >df1選擇(array_to_vector(“v1”)別名(“vec1”))收集()行(vec1 = DenseVector ([1.5, 2.5])))> > >df2=火花createDataFrame((((1.5,3.5),),,模式=“v1數組<飄>”)> > >df2選擇(array_to_vector(“v1”)別名(“vec1”))收集()行(vec1 = DenseVector ([1.5, 3.5])))> > >df3=火花createDataFrame((((1,3),),,模式=“v1數組< int >”)> > >df3選擇(array_to_vector(“v1”)別名(“vec1”))收集()行(vec1 = DenseVector ([1.0, 3.0])))