vector_norm

Возвращает норму Lp вектора с плавающей запятой с помощью указанной степени. Степень по умолчанию по умолчанию — 2,0 (норму Евклиида), если не указано.

Для соответствующей функции Databricks SQL смотрите функцию vector_norm.

Syntax

from pyspark.sql import functions as dbf

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

Parameters

Parameter	Тип	Описание
vector	pyspark.sql.Column или имя столбца	Входной векторный столбец.
degree	pyspark.sql.Column или имя столбца, необязательно	Степень норм (1.0 для L1, 2.0 для L2, float('inf') для бесконечности норм). По умолчанию — 2.0.

Returns

pyspark.sql.Column: норма Lp в виде значения с плавающей запятой.

Примеры

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