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 det elementmässigt medelvärdet av flyttalvektorer i en grupp. Alla vektorer måste ha samma dimension.
För motsvarande Databricks SQL-funktion, se vector_avg aggregeringsfunktionen.
Syntax
from pyspark.sql import functions as dbf
dbf.vector_avg(col=<col>)
Parameters
| Parameter | Type | Description |
|---|---|---|
col |
pyspark.sql.Column eller kolumnnamn |
Indatavektorkolumn. |
Returns
pyspark.sql.Column: Den elementmässigt genomsnittliga vektorn 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_avg('v')).first()[0]
# [2.0, 3.0]