Как сгенерировать n случайных целых чисел из k цифр

Question

Есть ли способ создать n случайных целых чисел из k цифр.

Например .. 2000 случайных числа, состоящие из [0, 2, 3]

Моего трика было использовать случайное число генератора, а затем присвоить значения, основанные на диапазонах?

Но было интересно, есть ли лучший способ сделать это в python?

Редактировать: Пример: [0,0,0, 2, 2,3,0,0,2,2, ..... 2000 элементов], содержащих 0,2 и 3 моего подход

def assign(x): 
     if x< 0.3: return 0 
     elif x<0.6: return 2 
     else: return 3 

    x = np.random.rand(num) 

    x = map(lamdba x:assign(x),x) 

Answer 1

Из звуков этого, это выглядит, как вы хотите, чтобы сформировать последовательность длины n, используя только найденные значения в списке k.

Функция Python random.choice в сочетании с пониманием списка идеально подходит для этого.

Следующая функция генерирует список длины n, при этом каждый элемент является случайным элементом, выбранным из k.

from random import choice 

def random_choices(n, k): 
    return [choice(k) for _ in xrange(n)] 

Это то же, что и простое понимание списка.

from random import choice 
foo = [choice(k) for _ in xrange(n)] 

* Благодаря Mr.goosberry за указание на то, что xrange следует заменить range в питона 3.x.x.

Answer 2

Теперь после вашего редактирования ясно, что вы хотите. Вы хотите дискретную выборку некоторых элементов внутри контейнера.

Просто подготовить классы и сделать это:

import numpy as np 
classes = [0, 2, 3] 
samples = np.random.choice(classes, 2000) 

Если вы хотите, чтобы некоторые конкретные вероятности:

import numpy as np 
classes = [0, 2, 3] 
samples = np.random.choice(classes, 2000, p=[0.3, 0.3, 0.4]) 

ВИДЕТЬ docs.

Реализация должна быть намного быстрее, чем ваш подход, который иногда называют дискретизацией рулетки или методом линейного поиска. Существуют некоторые возможные алгоритмы, упомянутые в wiki.

Answer 3

Вы можете достичь этого через list comprehension. Чтобы показать результат, я использую 20. Измените его на 2000 согласно вашему требованию.

>>> import random 
>>> x = 20 
>>> [random.choice([0, 2, 3]) for i in range(20)] 
[2, 2, 3, 2, 0, 2, 3, 2, 3, 3, 3, 0, 3, 2, 3, 2, 3, 2, 3, 2] 

Answer 4

Вы готовы использовать NumPy, поэтому я рекомендую вам использовать np.random.choice, то есть:

import numpy as np 

N = 2000 
print[np.random.choice([0, 2, 3], p=[1/3.0, 1/3.0, 1/3.0]) for x in range(N)] 

Answer 5

import numpy as np 
N = 10 
# Generate three vectors of n size 
zeros = np.zeros((N,), dtype = np.int) 
twos = np.zeros((N,), dtype = np.int) + 2 
threes = np.zeros((N,), dtype = np.int) + 3 
# Permutate all together 
df = [] 
df = np.append(df, [zeros, twos, threes]) 
df_shuffled = np.random.shuffle(df) 
print(df)