Skip to content

shuffle(col)

Описание

Функция shuffle() возвращает массив с элементами в случайном порядке. Порядок элементов меняется при каждом вызове функции.

Параметры

  • col: Column - массив для перемешивания

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

Array - массив с элементами в случайном порядке

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

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

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

# Перемешиваем массивы
df.select(
    "array_col",
    shuffle("array_col").alias("shuffled")
).show(truncate=False)

# Результат будет разным при каждом запуске:
# +------------+------------+
# |array_col   |shuffled    |
# +------------+------------+
# |[1, 2, 3, 4, 5]|[3, 1, 5, 2, 4]|
# |[a, b, c, d]|[c, a, d, b]|
# |null        |null        |
# +------------+------------+

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

df.select(
    "values",
    shuffle("values").alias("shuffled")
).show(truncate=False)

Примечания

  • Если массив NULL, возвращается NULL
  • Порядок элементов меняется при каждом вызове функции
  • Для сортировки массива используйте array_sort()
  • Для создания массива используйте array()
  • Для переворота массива используйте reverse()