vector_norm

Returnerar lp-normen för en flyttalvektor med den angivna graden. Grad är standardvärdet 2,0 (euklidisk norm) om det är ospecificerat.

För motsvarande Databricks SQL-funktion, se vector_norm funktion.

Syntax

from pyspark.sql import functions as dbf

dbf.vector_norm(vector=<vector>, degree=<degree>)

Parameters

Parameter Type Description
vector pyspark.sql.Column eller kolumnnamn Indatavektorkolumn.
degree pyspark.sql.Column eller kolumnnamn, valfritt Normgrad (1.0 för L1, 2.0 för L2, float('inf') för oändlighetsnorm). Standardinställningen är 2.0.

Returns

pyspark.sql.Column: Lp-normen som ett flyttalvärde.

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([([3.0, 4.0],)], schema)
df.select(dbf.vector_norm('v', dbf.lit(2.0).cast('float'))).first()[0]
# 5.0