transform_keys

Aplikuje funkci na každý pár klíč-hodnota v mapě a vrátí mapu s výsledky těchto aplikací jako nové klíče párů. Podporuje Spark Connect.

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

Syntaxe

from pyspark.sql import functions as dbf

dbf.transform_keys(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é klíče 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, {"foo": -2.0, "bar": 2.0})], ("id", "data"))
row = df.select(dbf.transform_keys(
    "data", lambda k, _: dbf.upper(k)).alias("data_upper")
).head()
sorted(row["data_upper"].items())

[('BAR', 2.0), ('FOO', -2.0)]