Случайный выбор в NumPy

Метод choice у генератора берёт значения из переданного массива или списка. Можно получить один элемент или сразу несколько. В этом уроке один и тот же элемент может попасть в результат больше одного раза.

Создадим генератор, набор букв и получим один символ. Выведем его:

import numpy as np

rng = np.random.default_rng()
pool = np.array(['a', 'b', 'c', 'd'])
pick = rng.choice(pool)
print(pick)

Теперь выберем три символа из того же набора. Выведем массив и его форму:

import numpy as np

rng = np.random.default_rng()
pool = np.array(['a', 'b', 'c', 'd'])
picks = rng.choice(pool, (3,))
print(picks)
print(picks.shape) # выведет (3,)

Создайте генератор и массив из чисел 1, 2, 3 и 4. Выберите из него один элемент и выведите его.

Из того же набора получите одномерный массив из 5 выбранных чисел. Выведите массив и его форму.