slice(x, start, length)

Описание

Функция slice() возвращает подмассив, начиная с указанного индекса и заданной длины. Индексы массива начинаются с 1, или с конца, если start отрицательный.

Параметры

x: Column - массив для среза
start: int - начальный индекс (1-базированный)
length: int - длина подмассива

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

Array - подмассив указанной длины

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

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

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

# Получаем срезы
df.select(
    "array_col",
    slice("array_col", 2, 3).alias("slice_2_3"),
    slice("array_col", -3, 2).alias("slice_neg3_2")
).show(truncate=False)

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

# Пример с разными длинами
df = spark.createDataFrame([
    ([10, 20, 30, 40],),
    ([100, 200, 300],)
], ["numbers"])

df.select(
    "numbers",
    slice("numbers", 1, 2).alias("first_two"),
    slice("numbers", -2, 2).alias("last_two")
).show(truncate=False)