transform_keys

Zastosowuje funkcję do każdej pary klucz-wartość w mapie i zwraca mapę, w której wyniki tych funkcji stają się nowymi kluczami dla par. Obsługuje program Spark Connect.

Aby uzyskać odpowiednią funkcję SQL usługi Databricks, zobacz transform_keys funkcja.

Składnia

from pyspark.databricks.sql import functions as dbf

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

Parametry

Parameter	Typ	Description
col	pyspark.sql.Column lub str	Nazwa kolumny lub wyrażenia.
f	function	Funkcja binarna.

Zwraca

pyspark.sql.Column: nowa mapa wpisów, w których obliczano nowe klucze, stosując daną funkcję do każdego argumentu wartości klucza.

Przykłady

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