numpy.random.RandomState.choice

метод

random.RandomState.choice(a, size=None, replace=True, p=None)

Генерирует случайную выборку из заданного одномерного массива

Новая в версии 1.7.0.

Примечание

Новый код должен использовать метод choice экземпляра default_rng(); см. Быстрое начало.

Параметры

a1-D array-like or int

Если массив NumPy, случайная выборка генерируется из его элементов. Если целое число, случайная выборка генерируется как если бы a был np.arange(a)

sizeint or tuple of ints, optional

Форма выходных данных. Если заданная форма, например, (m, n, k), то m * n * k элементов выбираются. По умолчанию None, в этом случае возвращается одно значение.

replaceboolean, optional

Выборка с или без возврата

p1-D array-like, optional

Вероятности, связанные с каждым элементом в a. Если не заданы, выборка предполагает равномерное распределение по всем элементам в a.

Возвращает

samplessingle item or ndarray

Сгенерированные случайные выборки

Возбуждает

ValueError

Если a является целым числом и меньше нуля, если a или p не одномерны, если a — массив-подобный нулевого размера, если p не вектор вероятностей, если a и p имеют разную длину или если replace=False, а размер выборки больше размера генеральной совокупности

См. также

randint, shuffle, permutation

Generator.choice

который следует использовать в новом коде

Примечания

Выборка случайных строк из двумерного массива с помощью этой функции невозможна, но возможна с помощью Generator.choice с помощью ключевого слова axis.

Примеры

Генерирует случайную равномерную выборку из np.arange(5) размера 3:

>>> np.random.choice(5, 3)
array([0, 3, 4]) # random
>>> #This is equivalent to np.random.randint(0,5,3)

Генерирует случайную неравномерную выборку из np.arange(5) размера 3:

>>> np.random.choice(5, 3, p=[0.1, 0, 0.3, 0.6, 0])
array([3, 3, 0]) # random

Генерирует случайную равномерную выборку из np.arange(5) размера 3 без возвращения:

>>> np.random.choice(5, 3, replace=False)
array([3,1,0]) # random
>>> #This is equivalent to np.random.permutation(np.arange(5))[:3]

Генерирует случайную неравномерную выборку из np.arange(5) размера 3 без возвращения:

>>> np.random.choice(5, 3, replace=False, p=[0.1, 0, 0.3, 0.6, 0])
array([2, 3, 0]) # random

Любое из вышеперечисленных можно повторить с произвольным массивоподобным объектом вместо только целых чисел. Например:

>>> aa_milne_arr = ['pooh', 'rabbit', 'piglet', 'Christopher']
>>> np.random.choice(aa_milne_arr, 5, p=[0.5, 0.1, 0.1, 0.3])
array(['pooh', 'pooh', 'pooh', 'Christopher', 'piglet'], # random
      dtype='<U11')