コード例 #1
0
ファイル: PostChargeTracker.py プロジェクト: jjf32897/YUSAG
def points_evaluator(rows):
    total_points = 0
    for row in rows:
        if not empty(row['points']):
            total_points += int(row['points'])
    return total_points
コード例 #2
0
ファイル: PostEventTracker.py プロジェクト: jjf32897/YUSAG
	def scorePerPossessionFullGame(self):
		'''
		Returns a dictionary where keys are game_ids and values are dictionaries of the form:
			{
				'home':{'team': TEAM_NAME, 'spp': SCORE_PER_POSSESSION},
				'away':{'team': TEAM_NAME, 'spp': SCORE_PER_POSSESSION}
			}
		'''
		returnDict = {}

		# game variables
		current_game = None
		game_counter = -1
		total_scores = None

		# possession variables
		current_possession_number = -1
		totaler = None
		current_possession = []

		with open(self.csvfile) as csvfile:
			reader = csv.DictReader(csvfile)
			for row in reader:

				# check if we are looking at a new game
				if row['game_id'] != current_game:

					# evaluate the last possession from the previous game
					if current_possession_number != -1:
						totaler['total'] += self.pe(current_possession)
						totaler['possessions'] += 1

					# add the game to the return dictionary
					if current_game != None:
						avg_scores = {'home':{}, 'away':{}}
						avg_scores['home']['team'] = total_scores['home']['team']
						avg_scores['away']['team'] = total_scores['away']['team']
						avg_scores['home']['spp'] = (total_scores['home']['total'] , total_scores['home']['possessions'])
						avg_scores['away']['spp'] = (total_scores['away']['total'] , total_scores['away']['possessions'])

						returnDict[current_game] = avg_scores

					# reset vars for new game
					current_game = row['game_id']
					game_counter += 1
					total_scores = {
							'home': {'team': None, 'total': 0, 'possessions': 0},
							'away': {'team': None, 'total': 0, 'possessions': 0}
						}
					teams = homeAwayLookup(current_game)
					total_scores['home']['team'] = teams['home']
					total_scores['away']['team'] = teams['away']
					current_possession_number = -1
					totaler = None
					current_possession = []

				# ignore lines without a possession
				if empty(row['possession_number']):
					continue

				# check if we have a new possession
				if row['possession_number'] != current_possession_number:
					# add score for the possession
					if current_possession_number != -1:
						totaler['total'] += self.pe(current_possession)
						totaler['possessions'] += 1

					# reset possession variables
					if total_scores['home']['team'] == row['possession_team']:
						totaler = total_scores['home']
					else:
						totaler = total_scores['away']
					current_possession_number = row['possession_number']
					current_possession = []

				current_possession.append(row)

		return returnDict
コード例 #3
0
ファイル: PostChargeTracker.py プロジェクト: mmenz/YUSAG
def points_evaluator(rows):
    total_points = 0
    for row in rows:
        if not empty(row["points"]):
            total_points += int(row["points"])
    return total_points
コード例 #4
0
ファイル: PostEventTracker.py プロジェクト: jjf32897/YUSAG
	def scorePerPossessionAfterEvent(self):
		'''
		Returns a dictionary where keys are game_ids and values are dictionaries of the form:
			{
				'home':{'team': TEAM_NAME, 'spp': {'own': SCORE_PER_POSSESSION, 'opp': SCORE_PER_POSSESSION}},
				'away':{'team': TEAM_NAME, 'spp': {'own': SCORE_PER_POSSESSION, 'opp': SCORE_PER_POSSESSION}}
			}
		'''
		returnDict = {}

		# game variables
		current_game = None
		game_counter = -1
		total_scores = None

		# possession variables
		current_possession_number = -1
		totaler = None
		current_possession = []
		event_counter = 0
		whose_event = None
		change_event_counter = False

		with open(self.csvfile) as csvfile:
			reader = csv.DictReader(csvfile)
			for row in reader:

				# check if we are looking at a new game
				if row['game_id'] != current_game:

					# evaluate the last possession from the previous game
					if current_possession_number != -1 and event_counter > 0:
						if whose_event == totaler['team']:
							totaler['total']['own'] += self.pe(current_possession)
							totaler['possessions']['own'] += 1
						else:
							totaler['total']['opp'] += self.pe(current_possession)
							totaler['possessions']['opp'] += 1
						event_counter -= 1

					# add the game to the return dictionary
					if current_game != None:
						avg_scores = {'home':{'spp':{}}, 'away':{'spp':{}}}
						avg_scores['home']['team'] = total_scores['home']['team']
						avg_scores['away']['team'] = total_scores['away']['team']
						if total_scores['home']['possessions']['own'] != 0:
							avg_scores['home']['spp']['own'] = (total_scores['home']['total']['own'] , total_scores['home']['possessions']['own'])
						else:
							avg_scores['home']['spp']['own'] = None

						if total_scores['home']['possessions']['opp'] != 0:
							avg_scores['home']['spp']['opp'] = (total_scores['home']['total']['opp'] , total_scores['home']['possessions']['opp'])
						else:
							avg_scores['home']['spp']['opp'] = None

						if total_scores['away']['possessions']['own'] != 0:
							avg_scores['away']['spp']['own'] = (total_scores['away']['total']['own'] , total_scores['away']['possessions']['own'])
						else:
							avg_scores['away']['spp']['own'] = None

						if total_scores['away']['possessions']['opp'] != 0:
							avg_scores['away']['spp']['opp'] = (total_scores['away']['total']['opp'] , total_scores['away']['possessions']['opp'])
						else:
							avg_scores['away']['spp']['opp'] = None

						returnDict[current_game] = avg_scores

					# reset vars for new game
					current_game = row['game_id']
					game_counter += 1
					total_scores = {
							'home': {'team': None, 'total': {'opp':0, 'own':0}, 'possessions': {'opp':0, 'own':0}},
							'away': {'team': None, 'total': {'opp':0, 'own':0}, 'possessions': {'opp':0, 'own':0}}
						}
					teams = homeAwayLookup(current_game)
					total_scores['home']['team'] = teams['home']
					total_scores['away']['team'] = teams['away']
					current_possession_number = -1
					totaler = None
					event_counter = 0
					current_possession = []
					whose_event = None

				# ignore lines without a possession
				if empty(row['possession_number']):
					continue

				# check if we have a new possession
				if row['possession_number'] != current_possession_number:
					# add score for the possession
					if current_possession_number != -1 and event_counter > 0 and current_possession != []:
						if whose_event == totaler['team']:
							totaler['total']['own'] += self.pe(current_possession)
							totaler['possessions']['own'] += 1
						else:
							totaler['total']['opp'] += self.pe(current_possession)
							totaler['possessions']['opp'] += 1
						event_counter -= 1

					# reset possession variables
					if total_scores['home']['team'] == row['possession_team']:
						totaler = total_scores['home']
					else:
						totaler = total_scores['away']
					current_possession_number = row['possession_number']
					current_possession = []

				if change_event_counter:
					event_counter = change_event_counter[0]
					whose_event = change_event_counter[1]
					change_event_counter = False

				# check if we have an event:
				if self.ef(row):
					change_event_counter = [self.ptt, row['possession_team']]
					
				current_possession.append(row)


		return returnDict