pyspark.sql.functions.map_keys

pyspark.sql.functions.map_keys(col: ColumnOrName) → pyspark.sql.column.Column

Collection function: Returns an unordered array containing the keys of the map.

Parameters
colColumn or str

name of column or expression

Examples

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