json_object_keys

Возвращает все ключи самого внешнего объекта JSON в виде массива. Если задан допустимый объект JSON, все ключи внешнего объекта будут возвращены в виде массива. Если это любая другая допустимая строка JSON, недопустимая строка JSON или пустая строка, функция возвращает значение NULL.

Синтаксис

from pyspark.sql import functions as sf

sf.json_object_keys(col)

Параметры

Параметр	Тип	Description
col	pyspark.sql.Column или str	Целевой столбец для вычисления.

Возвраты

pyspark.sql.Column: все ключи самого внешнего объекта JSON.

Примеры

from pyspark.sql import functions as sf
df = spark.createDataFrame([(None,), ('{}',), ('{"key1":1, "key2":2}',)], ['data'])
df.select(sf.json_object_keys(df.data).alias('r')).collect()

[Row(r=None), Row(r=[]), Row(r=['key1', 'key2'])]