try_element_at

Fungsi koleksi: Mengembalikan elemen array pada indeks atau nilai tertentu (berbasis 1) untuk kunci tertentu dalam peta. Untuk array, jika indeks adalah 0, Spark akan melemparkan kesalahan. Jika indeks < 0, mengakses elemen dari yang terakhir hingga yang pertama. Fungsi selalu mengembalikan NULL jika indeks melebihi panjang array. Untuk peta, fungsi selalu mengembalikan NULL jika kunci tidak terkandung dalam peta.

Untuk fungsi Databricks SQL yang sesuai, lihat try_element_at fungsi.

Syntax

from pyspark.databricks.sql import functions as dbf

dbf.try_element_at(col=<col>, extraction=<extraction>)

Parameter-parameternya

Pengaturan	Tipe	Description
col	pyspark.sql.Column atau str	Nama kolom yang berisi array atau peta.
extraction	pyspark.sql.Column atau str	Indeks yang akan diperiksa dalam array atau kunci untuk diperiksa di peta.

Examples

Contoh 1: Mendapatkan elemen pertama dari array

from pyspark.databricks.sql import functions as dbf
df = spark.createDataFrame([(["a", "b", "c"],)], ['data'])
df.select(dbf.try_element_at(df.data, dbf.lit(1))).show()

+-----------------------+
|try_element_at(data, 1)|
+-----------------------+
|                      a|
+-----------------------+

Contoh 2: Mendapatkan elemen terakhir dari array menggunakan indeks negatif

from pyspark.databricks.sql import functions as dbf
df = spark.createDataFrame([(["a", "b", "c"],)], ['data'])
df.select(dbf.try_element_at(df.data, dbf.lit(-1))).show()

+------------------------+
|try_element_at(data, -1)|
+------------------------+
|                       c|
+------------------------+