Partager via


transform_values

Applique une fonction à chaque paire clé-valeur dans une carte et retourne une carte avec les résultats de ces applications en tant que nouvelles valeurs pour les paires. Prend en charge Spark Connect.

Pour obtenir plus de détails sur la fonction SQL de Databricks correspondante, consultez transform_values.

Syntaxe

from pyspark.databricks.sql import functions as dbf

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

Paramètres

Paramètre Type Descriptif
col pyspark.sql.Column ou str Nom de la colonne ou de l’expression.
f function Fonction binaire.

Retours

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

Examples

from pyspark.databricks.sql import functions as dbf
df = spark.createDataFrame([(1, {"IT": 10.0, "SALES": 2.0, "OPS": 24.0})], ("id", "data"))
row = df.select(dbf.transform_values(
    "data", lambda k, v: dbf.when(k.isin("IT", "OPS"), v + 10.0).otherwise(v)
).alias("new_data")).head()
sorted(row["new_data"].items())
[('IT', 20.0), ('OPS', 34.0), ('SALES', 2.0)]