transform_values
Applique une fonction à chaque paire clé-valeur dans une carte et renvoie une carte avec les résultats de ces applications comme nouvelles valeurs pour les paires. Prend en charge Spark Connect.
Pour la fonction Databricks SQL correspondante, consultez la fonctiontransform_values.
Syntaxe
Python
from pyspark.sql import functions as dbf
dbf.transform_values(col=<col>, f=<f>)
parameter
parameter | Type | Description |
|---|---|---|
|
| Nom de la colonne ou de l'expression. |
|
| Une fonction binaire. |
Renvoie
pyspark.sql.Column: une nouvelle carte d'entrées où de nouvelles valeurs ont été calculées en appliquant la fonction donnée à chaque argument clé-valeur.
Exemples
Python
from pyspark.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())
Output
[('IT', 20.0), ('OPS', 34.0), ('SALES', 2.0)]