Метод shuffle класса RandomState

Метод shuffle класса RandomState перемешивает элементы массива или любой последовательности случайным образом. Метод изменяет исходный массив и не возвращает новый. Первым параметром метод принимает массив или последовательность (список, кортеж), которую нужно перемешать.

Синтаксис

random_state.shuffle(x)

Пример

Давайте создадим генератор случайных чисел и перемешаем одномерный массив:

import numpy as np

rng = np.random.RandomState(42)
arr = np.array([1, 2, 3, 4, 5, 6])
rng.shuffle(arr)
print(arr)

Результат выполнения кода:

[4 2 5 1 6 3]

Пример

Перемешаем список строк. Метод работает с любыми последовательностями:

import numpy as np

rng = np.random.RandomState(42)
data = ['a', 'b', 'c', 'd', 'e']
rng.shuffle(data)
print(data)

Результат выполнения кода:

['c', 'e', 'a', 'd', 'b']

Пример

Перемешаем многомерный массив. Метод перемешивает только первую ось массива:

import numpy as np

rng = np.random.RandomState(42)
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
rng.shuffle(arr)
print(arr)

Результат выполнения кода:

[[4 5 6]
 [7 8 9]
 [1 2 3]]

Пример

Метод shuffle не возвращает новый массив, а изменяет исходный. Попробуем сохранить результат в переменную:

import numpy as np

rng = np.random.RandomState(42)
arr = np.array([1, 2, 3, 4, 5])
res = rng.shuffle(arr)
print(res)
print(arr)

Результат выполнения кода:

None
[4 2 5 1 3]

Смотрите также

метод permutation,
который возвращает новый перемешанный массив
метод choice,
который генерирует случайную выборку из массива
метод rand,
который создает массив случайных чисел
класс RandomState,
который предоставляет генератор случайных чисел