#This program reads the data in AU gamedata.csv and calculates averages



import csv



def myavg(gamelist, games):

	score = 0

	for gamenumber in games:

		score += int(gamelist[gamenumber][3])	#get score from game in gamelist

	num = len(games)

	return score / num, num



#Read the data from the file to be read

reader = csv.reader(open('AUGamedata.csv', 'r'), delimiter=',')



conset, nonconset, tourneyset = set(), set(), set()

visitset, homeset = set(), set()

gamelist = []

i = 0

for row in reader:

	game = list(row)

	gamelist.append(game)

	

	if game[1] == 'C': conset.add(i)

	elif game[1] == 'N': nonconset.add(i)

	else: tourneyset.add(i)

	

	if game[2] == 'V': visitset.add(i)

	else: homeset.add(i)

	

	i +=1



convset = conset & visitset			#set intersections

conhset = conset & homeset

nonconhset = nonconset & homeset

nonconvset = nonconset & visitset

touvset = tourneyset & visitset

touhset = tourneyset & homeset

allset = conset | nonconset | tourneyset	#union of all sets



print('Average points per game')



print('Home team, nonconference: %4.2f \tNo of games: %s' % (myavg(gamelist, nonconhset)))

print('Home team, conference: %4.2f \t\tNo of games: %s' % (myavg(gamelist, conhset)))

print('Home team, tournament: %4.2f \t\tNo of games: %s'% (myavg(gamelist,touhset)))



print('Visiting team, nonconference: %4.2f \tNo of games: %s' % (myavg(gamelist, nonconvset)))

print('Visiting team, conference: %4.2f \tNo of games: %s' % (myavg(gamelist, convset)))

print('Visiting team, tournament: %4.2f \tNo of games: %s' % (myavg(gamelist, touvset)))



print('Overall home team average: %4.2f \tNo of games: %s' % (myavg(gamelist, homeset)))

print('Overall visiting team average: %4.2f \tNo of games: %s' % (myavg(gamelist, visitset)))

print('Overall average: %4.2f \t\t\tNo of games: %s' % (myavg(gamelist, allset)))