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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。

用法:

参数：

返回：

例子：