vector_l2_distance

Retourne la distance euclide (L2) entre deux vecteurs float. Les vecteurs doivent avoir la même dimension.

Pour obtenir plus de détails sur la fonction SQL de Databricks correspondante, consultez vector_l2_distance.

Syntax

from pyspark.sql import functions as dbf

dbf.vector_l2_distance(left=<left>, right=<right>)

Parameters

Parameter Type Description
left pyspark.sql.Column ou nom de colonne Première colonne vectorielle.
right pyspark.sql.Column ou nom de colonne Deuxième colonne vectorielle.

Returns

pyspark.sql.Column: distance L2 sous forme de valeur float.

Exemples

from pyspark.sql import functions as dbf
from pyspark.sql.types import ArrayType, FloatType, StructType, StructField

schema = StructType([StructField('a', ArrayType(FloatType())), StructField('b', ArrayType(FloatType()))])
df = spark.createDataFrame([([1.0, 2.0, 3.0], [4.0, 5.0, 6.0])], schema)
df.select(dbf.vector_l2_distance('a', 'b')).first()[0]
# 5.196152...