itertools бросает кубики: удваивает бросок дважды - PullRequest
Новая
       14

itertools бросает кубики: удваивает бросок дважды

0 голосов
/ 10 марта 2012

Я пытаюсь изучить библиотеку Python itertools, и я подумал, что хорошим тестом будет симуляция бросков кубиков. С помощью product легко сгенерировать все возможные броски и подсчитать количество возможных способов сделать это с помощью библиотеки collections. Я пытаюсь решить проблему, которая возникает в таких играх, как Монополия : когда выпадают двойные удары, вы бросаете снова, и ваш окончательный итог представляет собой сумму двух бросков.

Ниже приведена моя первая попытка решения проблемы: два счетчика, один для удвоений и другой для не удвоений. Я не уверен, есть ли хороший способ объединить их или два счетчика - даже лучший способ сделать это.

Я ищу хитрый способ решения (путем перечисления) проблемы броска кубиков с двойными числами с помощью itertools и коллекций. 

import numpy as np
from collections import Counter
from itertools import *

die_n = 2
max_num = 6

die = np.arange(1,max_num+1)
C0,C1  = Counter(), Counter()

for roll in product(die,repeat=die_n):
    if len(set(roll)) > 1: C0[sum(roll)] += 1
    else: C1[sum(roll)] += 1

1 Ответ

1 голос
/ 12 марта 2012

Оставляя здесь numpy ради простоты:

Сначала сгенерируйте все броски, будь то одиночные или двойные броски: 

from itertools import product
from collections import Counter

def enumerate_rolls(die_n=2, max_num=6):
    for roll in product(range(1, max_num + 1), repeat=die_n):
        if len(set(roll)) != 1:
            yield roll
        else:
            for second_roll in product(range(1, max_num + 1), repeat=die_n):
                yield roll + second_roll

Теперь несколько тестов: 

print(len(list(enumerate_rolls()))) # 36 + 6 * 36 - 6 = 246
A = list(enumerate_rolls(5, 4))
print(len(A)) # 4 ** 5 + 4 * 4 ** 5 - 4 = 5116
print(A[1020:1030]) # some double rolls (of five dice each!) and some single rolls

и результат: 

246
5116
[(1, 1, 1, 1, 1, 4, 4, 4, 4, 1), (1, 1, 1, 1, 1, 4, 4, 4, 4, 2), (1, 1, 1, 1, 1, 4, 4, 4, 4, 3), (1, 1, 1, 1, 1, 4, 4, 4, 4, 4), (1, 1, 1, 1, 2), (1, 1, 1, 1, 3), (1, 1, 1, 1, 4), (1, 1, 1, 2, 1), (1, 1, 1, 2, 2), (1, 1, 1, 2, 3)]

Чтобы получить итоги, используйте специальные возможности Counter: 

def total_counts(die_n=2, max_num=6):
    return Counter(map(sum, enumerate_rolls(die_n, max_num)))

print(total_counts())
print(total_counts(5, 4))

Результаты: 

Counter({11: 18, 13: 18, 14: 18, 15: 18, 12: 17, 16: 17, 9: 16, 10: 16, 17: 16, 18: 14, 8: 13, 7: 12, 19: 12, 20: 9, 6: 8, 5: 6, 21: 6, 22: 4, 4: 3, 3: 2, 23: 2, 24: 1})
Counter({16: 205, 17: 205, 18: 205, 19: 205, 21: 205, 22: 205, 23: 205, 24: 205, 26: 205, 27: 205, 28: 205, 29: 205, 25: 204, 20: 203, 30: 203, 15: 202, 14: 200, 31: 200, 13: 190, 32: 190, 12: 170, 33: 170, 11: 140, 34: 140, 35: 102, 10: 101, 9: 65, 36: 65, 8: 35, 37: 35, 7: 15, 38: 15, 6: 5, 39: 5, 40: 1})

Примечание: На данный момент нет способа вычислить вероятность для итогов.Вы должны знать, правильно ли это двойной рулон или общий рулон.

...