🔀 Как перемешать массив numpy: простой способ и советы

Для перемешивания массива Numpy можно использовать функцию shuffle() из модуля numpy.random.

Вот пример:

import numpy as np

array = np.array([1, 2, 3, 4, 5])
np.random.shuffle(array)
print(array)

Этот код перемешает элементы внутри массива в случайном порядке.

Как перемешать массив в NumPy?

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

Метод shuffle()

Первый способ перемешивания массива - использование метода shuffle(). Этот метод изменяет исходный массив, случайным образом перемешивая его элементы. Давайте рассмотрим пример:

import numpy as np

# Создаем массив
arr = np.array([1, 2, 3, 4, 5])

# Перемешиваем массив
np.random.shuffle(arr)

print(arr)
    

Вывод:

[4 2 5 1 3]

Как видно из примера, метод shuffle() перемешивает элементы массива в случайном порядке.

Функция permutation()

Еще один способ перемешивания массива - использование функции permutation(). Эта функция возвращает новый массив, в котором элементы исходного массива переставлены в случайном порядке. Рассмотрим пример:

import numpy as np

# Создаем массив
arr = np.array([1, 2, 3, 4, 5])

# Перемешиваем массив
shuffled_arr = np.random.permutation(arr)

print(shuffled_arr)
    

Вывод:

[2 1 3 5 4]

Функция permutation() возвращает новый массив, не изменяя исходный массив.

Метод choice()

Третий способ перемешивания массива - использование метода choice(). Этот метод возвращает новый массив, случайным образом выбирая элементы из исходного массива. Повторения элементов допускаются. Рассмотрим пример:

import numpy as np

# Создаем массив
arr = np.array([1, 2, 3, 4, 5])

# Перемешиваем массив
shuffled_arr = np.random.choice(arr, size=len(arr), replace=False)

print(shuffled_arr)
    

Вывод:

[3 4 2 5 1]

В примере мы использовали параметр replace=False, чтобы исключить повторяющиеся элементы в новом массиве. Если вы хотите разрешить повторения, установите replace=True.

Заключение

В этой статье мы рассмотрели три способа перемешивания массивов в NumPy. Метод shuffle() изменяет исходный массив, функция permutation() создает новый массив без изменения исходного, а метод choice() также создает новый массив, но с возможностью повторения элементов. Вы можете выбрать подходящий вариант в зависимости от ваших потребностей.