import random
import numpy as np

NUM_ROUNDS_PART_A = 60
max_ratio = 0.4
ratio = np.ones(NUM_ROUNDS_PART_A)
for i in range(1, NUM_ROUNDS_PART_A):
    ratio[i] = ratio[i - 1] + max_ratio/(NUM_ROUNDS_PART_A - 1)
np.random.shuffle(ratio)
NUM1 = np.zeros(NUM_ROUNDS_PART_A, dtype=int)
NUM2 = np.zeros(NUM_ROUNDS_PART_A, dtype=int)
for i in range(0, NUM_ROUNDS_PART_A):
    x = random.choice([0, 1])
    if x == 0:
        NUM1[i] = 120
        NUM2[i] = np.floor(ratio[i]*NUM1[i])
    else:
        NUM2[i] = 120
        NUM1[i] = np.floor(ratio[i]*NUM2[i])
NUMEROSITES = [NUM1, NUM2]