chaves_de_transformação

Aplica uma função a cada par key-valor em um mapa e retorna um mapa com os resultados dessas aplicações como a nova chave para os pares. Compatível com Spark Connect.

Para a função Databricks SQL correspondente, consulte a funçãotransform_keys.

Sintaxe

Python

from pyspark.databricks.sql import functions as dbf

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

Parâmetros

Parâmetro	Tipo	Descrição
col	pyspark.sql.Column ou str	Nome da coluna ou expressão.
f	function	Uma função binária.

Devoluções

pyspark.sql.Column: um novo mapa de entradas onde novas chaves foram calculadas aplicando a função fornecida a cada argumento de valor key .

Exemplos

Python

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

Output

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