multi-armed bandit simulation models in python