Bagikan melalui


transform_keys

Menerapkan fungsi ke setiap pasangan kunci-nilai dalam peta dan mengembalikan peta dengan hasil aplikasi tersebut sebagai kunci baru untuk pasangan. Mendukung Spark Connect.

Untuk fungsi Databricks SQL yang sesuai, lihat transform_keys fungsi.

Syntax

from pyspark.databricks.sql import functions as dbf

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

Parameter-parameternya

Pengaturan Tipe Description
col pyspark.sql.Column atau str Nama kolom atau ekspresi.
f function Fungsi biner.

Pengembalian Barang

pyspark.sql.Column: peta entri baru tempat kunci baru dihitung dengan menerapkan fungsi tertentu ke setiap argumen nilai kunci.

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