vector_norm

回傳使用指定度數的浮點向量的Lp範數。 若未指定,度數預設為 2.0(歐幾里得範數)。

關於對應的 Databricks SQL 函式,請參見 vector_norm 函數

語法

from pyspark.sql import functions as dbf

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

Parameters

參數 類型 Description
vector pyspark.sql.Column 或欄位名稱 輸入向量欄位。
degree pyspark.sql.Column 或欄位名稱,可選 範數度(1.0 L1 2.0 與 L2 float('inf') 無窮範數)。 預設為 2.0

Returns

pyspark.sql.Column:作為浮動值的 Lp 範數。

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