transform_keys

Applica una funzione a ogni coppia chiave-valore in una mappa e restituisce una mappa con i risultati di tali applicazioni come nuove chiavi per le coppie. Supporta Spark Connect.

Per la funzione SQL di Databricks corrispondente, vedere transform_keys funzione.

Sintassi

from pyspark.databricks.sql import functions as dbf

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

Parametri

Parametro	TIPO	Description
col	pyspark.sql.Column o str	Nome della colonna o dell'espressione.
f	function	Funzione binaria.

Restituzioni

pyspark.sql.Column: nuova mappa delle voci in cui sono state calcolate nuove chiavi applicando la funzione specificata a ogni argomento del valore della chiave.

Esempi

from pyspark.databricks.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)]