Генерация случайной перестановки с помощью numpy

Функция np.random.permutation() используется для генерации случайной перестановки указанного массива. Она случайным образом перемешивает элементы массива и возвращает новый массив с такой перестановкой.

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

python
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
permuted_arr = np.random.permutation(arr)
print(permuted_arr)

В данном примере создаётся массив arr, содержащий числа от 1 до 5. Затем с помощью функции np.random.permutation() создаётся новый массив permuted_arr, в котором элементы arr перемешаны случайным образом. Наконец, этот массив выводится на экран.

Пример вывода:

[2 3 1 5 4]

Заметьте, что каждый раз при запуске этого кода будет выводиться разная перестановка элементов arr.

Также функция np.random.permutation() может применяться к многомерным массивам. В этом случае она будет случайно перемешивать только первое измерение массива, сохраняя относительный порядок элементов в остальных измерениях. Например:

python
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
permuted_arr = np.random.permutation(arr)
print(permuted_arr)

Вывод:

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

Здесь массив arr имеет размерность (3, 3), то есть содержит 3 строки и 3 столбца. После применения функции np.random.permutation() элементы первой строки перемешались, а остальные строки остались на своих местах. В результате получился массив permuted_arr с новой перестановкой элементов первой строки.