Question

You will now do the same thing you did in the previous problems, but with a new experiment: instead of rolling one die and recording the value, you will simulate rolling ?n dies and recording their sum. For example, if n=2 and the first die shows up as a 3, and the second die shows up as a 1, the sum (and the value we record) would be 4.

TO DO: Complete the two function stubs below and then demonstrate by providing code which would print out the probability distribution for rolling 2 dice 100,000 times.

Hint: Not required, but think about how you might do this in one line using Numpy and list comprehensions.

import random
seed(0)
  
def roll_and_add_dice(num_dice, num_trials = 10**6):
pass # Your code here

Answer 1

import numpy
import matplotlib.pyplot as plt

def roll_and_add_dice(num_dice, num_trials = 10**5):
    return [numpy.random.randint(0, 7, num_dice).sum() for i in range(num_trials) ]

if __name__ == "__main__":
    results = roll_and_add_dice(2)
    print(numpy.bincount(results))
    plt.stem(numpy.bincount(results), linefmt='-.')
    plt.margins(0.1)
    plt.show()