coinsim.py

import numpy as np


day2 = []
for i in range(10000):
    
    # Randomly select coin 1 or 2 on day 1
    X = np.random.randint(0, 2)
    
    # Flip either coin and append the outcome to day2 vector
    if X == 1:
        day2.append(np.random.choice([0, 1], p=[.3, .7]))
    else:
        day2.append(np.random.choice([0, 1], p=[.4, .6]))

print(np.mean(day2))