vector_norm

Devolve a norma Lp de um vetor flutuante usando o grau especificado. O grau por defeito é 2.0 (norma euclidiana) se não especificado.

Para a função SQL do Databricks correspondente, veja vector_norm função.

Sintaxe

from pyspark.sql import functions as dbf

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

Parameters

Parâmetro Tipo Description
vector pyspark.sql.Column ou nome da coluna Coluna de vetores de entrada.
degree pyspark.sql.Column ou nome da coluna, opcional Grau de norma (1.0 para L1, 2.0 para L2, float('inf') para norma infinita). O valor padrão é 2.0.

Returns

pyspark.sql.Column: A norma Lp como valor flutuante.

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