import random

def infect():
    infected = random.sample(range(1,7),2)
    for j in range(0,5):
        k = random.randrange(1,7)
        if k in infected:
            return infected
        infected.append(k)
    return infected

def run_sim(n):
    sim = []
    cnt = 0
    for i in range(0,n):
        sim = infect()
        if len(sim) > 3:
            cnt = cnt+1
    return cnt/n

import numpy as np
n_infected = []
n = 10000
for i in range(n):
    infected = [True, False, False, False, False, False]

    x = np.random.randint(1, len(infected))
    infected[x] = True
    while True:
        x = np.random.randint(1, len(infected))
        if infected[x] is True:
            break
        infected[x] = True
    n_infected.append(sum(infected))


sum([1 for inf in n_infected if inf > 3]) / n