CS110 4.2 Preclass

hat.py

import random
import numpy as np


def hatproblem(people):
    hats = [i for i in range(1,people+1)]
    hatsreplica = [i for i in range(1,people+1)]
    random.shuffle(hatsreplica)
    count = 0
    for i in range(len(hats)):
        if hats[i] == hatsreplica[i]:
            count += 1
    return count

list = []
for i in range(1000):
    list.append(hatproblem(i))

print(np.mean(list))