json_object_keys(json_object)

Описание

Функция json_object_keys() возвращает массив ключей JSON-объекта. Если входное значение не является JSON-объектом, возвращается NULL.

Параметры

json_object: Column - JSON-объект в виде строки

Возвращаемое значение

Array[String] - массив ключей JSON-объекта или NULL

Пример использования

from pyspark.sql import SparkSession
from pyspark.sql.functions import json_object_keys

# Создаем SparkSession
spark = SparkSession.builder.appName("json_object_keys_example").getOrCreate()

# Создаем DataFrame с JSON-объектами
data = [
    (1, '{"name": "Alice", "age": 25}'),
    (2, '{"city": "New York", "country": "USA"}'),
    (3, '{}'),
    (4, '["a", "b", "c"]'),
    (5, None)
]
df = spark.createDataFrame(data, ["id", "json"])

# Получаем ключи JSON-объектов
result = df.select(
    "id",
    json_object_keys("json").alias("keys")
).show()

# Результат:
# +---+------------------+
# | id|             keys|
# +---+------------------+
# |  1|[name, age]      |
# |  2|[city, country]  |
# |  3|[]               |
# |  4|             null|
# |  5|             null|
# +---+------------------+