map_keys
Retorna um array não ordenado contendo a chave do mapa.
Sintaxe
Python
from pyspark.sql import functions as sf
sf.map_keys(col)
Parâmetros
Parâmetro | Tipo | Descrição |
|---|---|---|
|
| Nome da coluna ou expressão |
Devoluções
pyspark.sql.Column: chave do mapa como uma matriz.
Exemplos
Exemplo 1 : Extraindo a chave de um mapa simples
Python
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()
Output
+--------------------------------+
|sort_array(map_keys(data), true)|
+--------------------------------+
| [1, 2]|
+--------------------------------+
Exemplo 2 : Extraindo a chave de um mapa com chave complexa
Python
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()
Output
+--------------------------------+
|sort_array(map_keys(data), true)|
+--------------------------------+
| [[1, 2], [3, 4]]|
+--------------------------------+
Exemplo 3 : Extraindo a chave de um mapa vazio
Python
from pyspark.sql import functions as sf
df = spark.sql("SELECT map() as data")
df.select(sf.map_keys("data")).show()
Output
+--------------+
|map_keys(data)|
+--------------+
| []|
+--------------+