聚合函數：回傳群中浮點向量的元素總和。 所有向量的維度必須相同。
關於對應的 Databricks SQL 函式，請參見 vector_sum aggregate 函數。
語法
from pyspark.sql import functions as dbf
dbf.vector_sum(col=<col>)
Parameters
| 參數 | 類型 | Description |
|---|---|---|
col |
pyspark.sql.Column 或欄位名稱 |
輸入向量欄位。 |
Returns
pyspark.sql.Column：元素層級的求和向量，作為一個浮點陣列。
Examples
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]