Question

I was wondering if someone can clarify this line for me.

Create a function die(x) which rolls a
die x times keeping track of how many
times each face comes up and returns a
1X6 array containing these numbers.

I am not sure what this means when it says 1X6 array ? I am using the randint function from numpy so the output is already an array (or list) im not sure.

Thanks

Answer 1

def die(x):
    return np.bincount(np.random.random_integers(0, 5, size=x))

np.random.random_integers(0,5,size=x) rolls the die x times (faces are represented by numbers from 0 to 5 including).

np.bincount() returns the number of occurrences of each value in the array i.e., how many times each face comes up.

Example

>>> a = np.random.random_integers(0, 5, size=10)
>>> a
array([3, 5, 0, 5, 0, 5, 5, 1, 3, 0])
>>> np.bincount(a)
array([3, 1, 0, 2, 0, 4])