array_append(col, value)

Описание

Функция array_append() возвращает массив элементов из col вместе с добавленным элементом value в конец массива.

Параметры

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

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

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

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

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

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

# Добавляем элементы в конец массива
df.select(
    "array_col",
    array_append("array_col", 4).alias("appended_number"),
    array_append("array_col", "d").alias("appended_letter")
).show(truncate=False)

# Результат:
# +---------+---------------+---------------+
# |array_col|appended_number|appended_letter|
# +---------+---------------+---------------+
# |[1, 2, 3]|[1, 2, 3, 4]  |[1, 2, 3, d]  |
# |[a, b, c]|[a, b, c, 4]  |[a, b, c, d]  |
# |null     |null          |null          |
# +---------+---------------+---------------+

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

df.select(
    "values",
    array_append("values", 3.0).alias("appended_float"),
    array_append("values", True).alias("appended_bool")
).show(truncate=False)