Freigeben über


transform_keys

Wendet eine Funktion auf jedes Schlüssel-Wert-Paar in einer Karte an und gibt eine Karte mit den Ergebnissen dieser Anwendungen als neue Schlüssel für die Paare zurück. Unterstützt Spark Connect.

Die entsprechende Databricks SQL-Funktion finden Sie unter transform_keys Funktion.

Syntax

from pyspark.databricks.sql import functions as dbf

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

Die Parameter

Parameter Typ Description
col pyspark.sql.Column oder str Name der Spalte oder des Ausdrucks.
f function Eine binäre Funktion.

Rückkehr

pyspark.sql.Column: eine neue Zuordnung von Einträgen, bei denen neue Schlüssel durch Anwenden einer bestimmten Funktion auf jedes Schlüsselwertargument berechnet wurden.

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