Kommentar
Åtkomst till den här sidan kräver auktorisering. Du kan prova att logga in eller ändra kataloger.
Åtkomst till den här sidan kräver auktorisering. Du kan prova att ändra kataloger.
Mängdfunktion: returnerar den elementvisa summan av flyttalvektorer i en grupp. Alla vektorer måste ha samma dimension.
För motsvarande Databricks SQL-funktion, se vector_sum aggregeringsfunktionen.
Syntax
from pyspark.sql import functions as dbf
dbf.vector_sum(col=<col>)
Parameters
| Parameter | Type | Description |
|---|---|---|
col |
pyspark.sql.Column eller kolumnnamn |
Indatavektorkolumn. |
Returns
pyspark.sql.Column: Den elementvisa sumvektorn som en matris med flyttal.
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]