Partager via


map_keys

Retourne un tableau non ordonné contenant les clés de la carte.

Syntaxe

from pyspark.sql import functions as sf

sf.map_keys(col)

Paramètres

Paramètre Type Descriptif
col pyspark.sql.Column ou str Nom de la colonne ou de l’expression

Retours

pyspark.sql.Column: clés de la carte sous forme de tableau.

Examples

Exemple 1 : Extraction de clés à partir d’une carte simple

from pyspark.sql import functions as sf
df = spark.sql("SELECT map(1, 'a', 2, 'b') as data")
df.select(sf.sort_array(sf.map_keys("data"))).show()
+--------------------------------+
|sort_array(map_keys(data), true)|
+--------------------------------+
|                          [1, 2]|
+--------------------------------+

Exemple 2 : Extraction de clés à partir d’une carte avec des clés complexes

from pyspark.sql import functions as sf
df = spark.sql("SELECT map(array(1, 2), 'a', array(3, 4), 'b') as data")
df.select(sf.sort_array(sf.map_keys("data"))).show()
+--------------------------------+
|sort_array(map_keys(data), true)|
+--------------------------------+
|                [[1, 2], [3, 4]]|
+--------------------------------+

Exemple 3 : Extraction de clés à partir d’une carte vide

from pyspark.sql import functions as sf
df = spark.sql("SELECT map() as data")
df.select(sf.map_keys("data")).show()
+--------------+
|map_keys(data)|
+--------------+
|            []|
+--------------+