def plotDistanceByTime(player):
XINTERVALLENGTH = 5.0
XMAX = 60.0
YINTERVALLENGTH = 3.0
YMAX = 30.0
MARKERSIZE = 5000
(xMids, yMids, buckets) = plotShots.makeBuckets(XINTERVALLENGTH, YINTERVALLENGTH, XMAX, YMAX)
maxTotal = 0
# Each bucket contains two elements
# Number of made shots
# Total number of shots
for shot in player.made:
if shot.time >= XMAX:
xInt = len(buckets) - 1
else:
xInt = math.floor(shot.time / XINTERVALLENGTH)
if shot.distance >= YMAX:
yInt = len(buckets[0]) - 1
else:
yInt = math.floor(shot.distance / YINTERVALLENGTH)
xInt = int(xInt)
yInt = int(yInt)
buckets[xInt][yInt][0] += 1
buckets[xInt][yInt][1] += 1
if buckets[xInt][yInt][1] > maxTotal:
maxTotal = buckets[xInt][yInt][1]
for shot in player.missed:
if shot.time >= XMAX:
xInt = len(buckets) - 1
else:
xInt = math.floor(shot.time / XINTERVALLENGTH)
if shot.distance >= YMAX:
yInt = len(buckets[0]) - 1
else:
yInt = math.floor(shot.distance / YINTERVALLENGTH)
xInt = int(xInt)
yInt = int(yInt)
buckets[xInt][yInt][1] += 1
if buckets[xInt][yInt][1] > maxTotal:
maxTotal = buckets[xInt][yInt][1]
percentages = []
shotTotals = []
# Find Field Goal Percentage for Each Bucket
# First Element of the bucket = FG%
# Add the FG% to the percentages list
# Add normalised shot total to the shotTotals list
for xInt in buckets:
for bucket in xInt:
if bucket[1] != 0:
bucket[0] = float(bucket[0]) / bucket[1]
# normTotal = (float(bucket[1]) / maxTotal) * MARKERSIZE
normTotal = (float(bucket[1]) / maxTotal) ** 2 * MARKERSIZE
percentages.append(bucket[0])
shotTotals.append(normTotal)
# The actual points plotted are the centers of each bucket
# yMid for every xMid
xVals = []
yVals = []
for x in xMids:
xValues = [x] * len(yMids)
xVals.extend(xValues)
yVals.extend(yMids)
# Get a colourmap for the FG percentages
# Percentages are reflected via the gradient of the marker
# The darker it is, the higher the percentage
colourMap = cm.jet
figure()
sct = scatter(xVals, yVals, s=shotTotals, c=percentages, cmap=colourMap)
sct.set_alpha(0.75)
lims = xlim()
xlim(-5, lims[1])
xlabel("Elapsed Game Time", fontsize = 35)
ylabel("Distance From Basket", fontsize = 35)
suptitle("Shot Distance vs Game Time for " + str(player.name), fontsize = 45)
cb = colorbar()
cb.set_label("FG %", labelpad = 5, fontsize = 25)
cb.set_ticks([0, 1.00])
show()
def plotDistanceBySituation(player):
XINTERVALLENGTH = 0.5
XMIN = -2.0
XMAX = 5.5
YINTERVALLENGTH = 3.0
YMAX = 30.0
MARKERSIZE = 4000
(xMids, yMids, buckets) = plotShots.makeBuckets(XINTERVALLENGTH, YINTERVALLENGTH, XMAX, YMAX, XMIN)
maxTotal = 0
for shot in player.made:
score = shot.score.split('-')
awayScore = float(score[0])
homeScore = float(score[1])
sit = situation(homeScore, awayScore)
for i in range(len(xMids)):
if sit < (XMIN + (i + 1) * XINTERVALLENGTH):
xInt = i
break
if shot.distance >= YMAX:
yInt = len(buckets[0]) - 1
else:
yInt = math.floor(shot.distance / YINTERVALLENGTH)
xInt = int(xInt)
yInt = int(yInt)
buckets[xInt][yInt][0] += 1
buckets[xInt][yInt][1] += 1
if buckets[xInt][yInt][1] > maxTotal:
maxTotal = buckets[xInt][yInt][1]
for shot in player.missed:
score = shot.score.split('-')
awayScore = float(score[0])
homeScore = float(score[1])
sit = situation(homeScore, awayScore)
for i in range(len(xMids)):
if sit < (XMIN + (i + 1) * XINTERVALLENGTH):
xInt = i
break
if shot.distance >= YMAX:
yInt = len(buckets[0]) - 1
else:
yInt = math.floor(shot.distance / YINTERVALLENGTH)
xInt = int(xInt)
yInt = int(yInt)
buckets[xInt][yInt][1] += 1
if buckets[xInt][yInt][1] > maxTotal:
maxTotal = buckets[xInt][yInt][1]
percentages = []
shotTotals = []
# Find Field Goal Percentage for Each Bucket
# First Element of the bucket = FG%
# Add the FG% to the percentages list
# Add normalised shot total to the shotTotals list
for xInt in buckets:
for bucket in xInt:
if bucket[1] != 0:
bucket[0] = float(bucket[0]) / bucket[1]
# normTotal = (float(bucket[1]) / maxTotal) * MARKERSIZE
normTotal = (float(bucket[1]) / maxTotal) ** 2 * MARKERSIZE
percentages.append(bucket[0])
shotTotals.append(normTotal)
# The actual points plotted are the centers of each bucket
# yMid for every xMid
xVals = []
yVals = []
for x in xMids:
xValues = [x] * len(yMids)
xVals.extend(xValues)
yVals.extend(yMids)
# Get a colourmap for the FG percentages
# Percentages are reflected via the gradient of the marker
# The darker it is, the higher the percentage
colourMap = cm.jet
figure()
sct = scatter(xVals, yVals, s=shotTotals, c=percentages, cmap=colourMap)
sct.set_alpha(0.75)
xlabel("Situation Factor", fontsize = 35)
lims = xlim()
xlim(XMIN - 0.5, XMAX + 0.5)
ylabel("Distance From Basket", fontsize = 35)
suptitle("Shot Distance vs Situation Factor for " + str(player.name), fontsize = 45)
cb = colorbar(sct)
cb.set_label("FG %", labelpad = 5, fontsize = 25)
cb.set_ticks([0, 1])
cb.set_ticklabels(['0', '1'])
show()