Изменить вероятность выпадения чисел?

Question

[Алексей Орлов]

Делаю первые шаги в изучении python. Установил модуль numPy.
У меня выводится любое случайное число до 100 при помощи
np.random.randint(100, size=1)

Вопрос: Возможно ли сделать так, чтоб вероятность вывода значений от 0 до 50 была выше, нежели от 50 до 100?

Буду благодарен, если напишите в сторону чего смотреть, может скинете какой-либо источник, где будет подробно описан этот процесс.

Answer 1

[Владимир Проскурин]

Нужно смотреть на результат rand как на некое число, вероятность выпадания которого равно 1 / N. И чтобы ваше реальное число выпадало в два раза чаще, ее вероятность должна быть равна двум этим числам 2 * (1 / N).
В вашем примере, вероятность выпадания любого числа равна 1 / 100, чтобы для чисел от 0 до 50 вероятность была например 2 / N, нужно чтобы они занимали по два числа из результата randint.

rand = np.random.randint(150, size=1)[0]
if rand < 100:
    res = rand // 2
else:
    res = rand - 50

т.е. если rand < 100, то это числа первой половины, вероятность каждого из которой равны 2 / N, а для rand > 100 and rand < 150 все также 1 / N

Comments

[dmshar]

Извините, но по вашей формуле числа в диапазоне от 50 до 100 вообще никогда не будут генерироваться. Только от 0 до 50 и от 100 до 150.

[Алексей Орлов]

Очень круто, спасибо!

[Владимир Проскурин]

dmshar, действительно, забыл вычесть 50. Спасибо что заметили.

[Алексей Орлов]

Владимир Проскурин, не очень понятно, например выпало число 101, получается оно следует по второму условию и получается 1, тоесть максимально возможным значением же будет 50, если выпадет 150 и из него - 100 или я что-то не так понимаю?

[Владимир Проскурин]

Алексей Орлов, да, там нужно 50 вычитать, тогда получится 149 - 50 = 99. Поправил код.

[dmshar]

Владимир Проскурин,

Вероятность получения исходных чисел в диапазоне от 0 до 150 - равное, т.е. каждое по 1/150. Вероятность появления ЛЮБОГО числа при этом в диапазоне от 1 до 100 равно 100/150=2/3. А любого числа в диапазоне от 100 до 150 = 50/150=1/3. После вычитания 50 вы получили, что числа от 0 до 50 у вас генерируются с вероятностью 2/3, а от 50 до 100 - 1/3.
А соответственно, каждого числа до 50 - 2/150, а каждого числа после 50 1/150. ( а не 2/100 и 1/100). Не знаю, то-ли это решение, которое ожидал ТС.

Answer 2

[dmshar]

Начните с "смотрения" в сторону теории вероятности. Для начала. Источников в сети - сотни и тысячи.

Теперь про ваш вопрос.
Вы взяли РАВНОМЕРНОЕ распределение. И генерируете числа. Функции для генерации чисел согласно основным законом распределения действительно реализованы в Python/numpy. Если вы хотите генерировать числа с неким распределением, которое вы сами зададите - ничего не мешает это делать. Например, взяв за основу то самое равномерное распределение и выполнив преобразование по формуле плотности распределения, которую сами и напишете.