transform_keys

Aplica una función a cada par clave-valor de un mapa y devuelve un mapa con los resultados de esas aplicaciones como claves nuevas para los pares. Soporta Spark Connect.

Para obtener la función SQL de Databricks correspondiente, consulte transform_keys function.

Syntax

from pyspark.databricks.sql import functions as dbf

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

Parámetros

Parámetro	Tipo	Description
col	pyspark.sql.Column o str	Nombre de columna o expresión.
f	function	Una función binaria.

Devoluciones

pyspark.sql.Column: un nuevo mapa de entradas en las que se calcularon nuevas claves aplicando la función dada a cada argumento de valor de clave.

Examples

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)]