transform_values

Použije funkci na každý pár klíč-hodnota v mapě a vrátí mapu s výsledky použití těchto funkcí jako nové hodnoty pro páry. Podporuje Spark Connect.

Odpovídající funkci SQL Databricks vizte funkcetransform_values.

Syntaxe

from pyspark.sql import functions as dbf

dbf.transform_values(col=<col>, f=<f>)

Parametry

Parameter	Typ	Description
col	pyspark.sql.Column nebo str	Název sloupce nebo výrazu
f	function	Binární funkce.

Návraty

pyspark.sql.Column: Nová mapa položek, kde byly nové hodnoty vypočítány použitím dané funkce pro každý argument hodnoty klíče.

Examples

from pyspark.sql import functions as dbf
df = spark.createDataFrame([(1, {"IT": 10.0, "SALES": 2.0, "OPS": 24.0})], ("id", "data"))
row = df.select(dbf.transform_values(
    "data", lambda k, v: dbf.when(k.isin("IT", "OPS"), v + 10.0).otherwise(v)
).alias("new_data")).head()
sorted(row["new_data"].items())

[('IT', 20.0), ('OPS', 34.0), ('SALES', 2.0)]