Upper Confidence Bound (UCB) - Upper Confidence Bound in Python - Step 1