聚合函数：返回组中浮向量元素的和。 所有向量必须具有相同的维度。
有关相应的 Databricks SQL 函数，请参阅 vector_sum 聚合函数。
Syntax
from pyspark.sql import functions as dbf
dbf.vector_sum(col=<col>)
Parameters
| 参数 | 类型 | 说明 |
|---|---|---|
col |
pyspark.sql.Column 或列名 |
输入向量列。 |
Returns
pyspark.sql.Column：以元素方式求和向量作为浮点数组。
示例
from pyspark.sql import functions as dbf
from pyspark.sql.types import ArrayType, FloatType, StructType, StructField
schema = StructType([StructField('v', ArrayType(FloatType()))])
df = spark.createDataFrame([([1.0, 2.0],), ([3.0, 4.0],)], schema)
df.select(dbf.vector_sum('v')).first()[0]
# [4.0, 6.0]