CS110 4.2 Preclass

hire.py

import random
import math
import matplotlib.pyplot as plt
import numpy as np

masterinterview = []
masterhire = []
for apps in range(1,1501):
    interviewlist = []
    hirelist = []
    for reps in range(100):
        interview = 0   #count interviews
        hire = 0   #count hires
        best = 0   #current worker's "strength"
        cand = [random.random() for i in range(1,apps+1)]   #list of candidates with "strengths" from 1 to 100
        for i in range(len(cand)):
            interview += 1   #count interviews
            if cand[i] > best:   #if candidate is better than current worker, hire
                best = cand[i]
                hire += 1   #count hire
        interviewlist.append(interview)
        hirelist.append(hire)
    masterinterview.append(int(np.mean(interviewlist)))
    masterhire.append(np.mean(hirelist))

plt.ylabel("Number of hires")
plt.xlabel("Number of candidates")
plt.plot([i for i in range(1,1501)],[math.log(i) for i in range(1,1501)], label = "theoretical estimate")
plt.scatter([i for i in range(1,1501)],masterhire, color = "r", s = 0.4, label = "observed estimate")
plt.legend(loc="lower right")
plt.show()