array_prepend(col, value)

Описание

Функция array_prepend() возвращает массив, содержащий указанный элемент, а также все элементы из исходного массива. Элемент добавляется в начало массива.

Параметры

col: Column - исходный массив
value: Any - значение для добавления в начало массива

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

Array - новый массив с добавленным элементом в начале

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

from pyspark.sql.functions import array_prepend, array, lit

# Создаем DataFrame с массивами
df = spark.createDataFrame([
    ([1, 2, 3],),
    (["a", "b", "c"],),
    (None,)
], ["array_col"])

# Добавляем элементы в начало массива
df.select(
    "array_col",
    array_prepend("array_col", 0).alias("prepended_number"),
    array_prepend("array_col", "x").alias("prepended_letter")
).show(truncate=False)

# Результат:
# +---------+----------------+----------------+
# |array_col|prepended_number|prepended_letter|
# +---------+----------------+----------------+
# |[1, 2, 3]|[0, 1, 2, 3]   |[x, 1, 2, 3]   |
# |[a, b, c]|[0, a, b, c]   |[x, a, b, c]   |
# |null     |null           |null           |
# +---------+----------------+----------------+

# Пример с разными типами данных
df = spark.createDataFrame([
    ([1.0, 2.0],),
    ([True, False],),
    ([None, 1],)
], ["values"])

df.select(
    "values",
    array_prepend("values", 0.0).alias("prepended_float"),
    array_prepend("values", True).alias("prepended_bool")
).show(truncate=False)