transform_keys

Применяет функцию к каждой паре "ключ-значение" в карте и возвращает карту с результатами этих приложений в качестве новых ключей для пар. Поддерживает Spark Connect.

Для соответствующей функции Databricks SQL смотрите функцию transform_keys.

Синтаксис

from pyspark.databricks.sql import functions as dbf

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

Параметры

Параметр	Тип	Description
col	pyspark.sql.Column или str	Имя столбца или выражения.
f	function	Двоичная функция.

Возвраты

pyspark.sql.Column: новая карта записей, в которых были вычисляются новые ключи путем применения данной функции к каждому аргументу значения ключа.

Примеры

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