Win the Bandid Game! https://github.com/andyzeng1980/Data_Science_Workshop/tree/master/1-intro

game.py

import operator

from BanditGame import BanditGame
ba = BanditGame()

results = {}
for arm in range(1, 6):
	results[arm] = 0
	for game in range(20):	
		startValue = ba.money()
		ba.play(arm)
		results[arm] = results[arm] + ba.money()-startValue #how much money you'll gane with this arm

bestArm = max(results.iteritems(), key=operator.itemgetter(1))[0]

while ba._money_check:
	ba.play(bestArm)