Python pyspark array_to_vector用法及代碼示例

本文簡要介紹 pyspark.ml.functions.array_to_vector 的用法。

用法: pyspark.ml.functions.array_to_vector(col)

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

版本 3.1.0 中的新函數。

參數：

col： pyspark.sql.Column 或 str: 輸入欄

Column: 密集向量的轉換列。

例子：

>>> from pyspark.ml.functions import array_to_vector
>>> df1 = spark.createDataFrame([([1.5, 2.5],),], schema='v1 array<double>')
>>> df1.select(array_to_vector('v1').alias('vec1')).collect()
[Row(vec1=DenseVector([1.5, 2.5]))]
>>> df2 = spark.createDataFrame([([1.5, 3.5],),], schema='v1 array<float>')
>>> df2.select(array_to_vector('v1').alias('vec1')).collect()
[Row(vec1=DenseVector([1.5, 3.5]))]
>>> df3 = spark.createDataFrame([([1, 3],),], schema='v1 array<int>')
>>> df3.select(array_to_vector('v1').alias('vec1')).collect()
[Row(vec1=DenseVector([1.0, 3.0]))]

相關用法

注：本文由純淨天空篩選整理自spark.apache.org大神的英文原創作品 pyspark.ml.functions.array_to_vector。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

例子：