transform_keys

Applique une fonction à chaque paire clé-valeur dans un dictionnaire et renvoie un dictionnaire avec les résultats de ces applications comme nouvelles clés pour les paires. Prend en charge Spark Connect.

Pour la fonction Databricks SQL correspondante, consultez la fonctiontransform_keys.

Syntaxe

Python

from pyspark.sql import functions as dbf

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

parameter

parameter	Type	Description
col	pyspark.sql.Column OU str	Nom de la colonne ou de l'expression.
f	function	Une fonction binaire.

parameter	Type	Description
col	pyspark.sql.Column OU str	Nom de la colonne ou de l'expression.
f	function	Une fonction binaire.

Renvoie

pyspark.sql.Column: une nouvelle carte d’entrées où de nouvelles clés ont été calculées en appliquant une fonction donnée à chaque argument de valeur de clé.

Exemples

Python

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

Output

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