INNER CODE UNIT · Python
y_pos
chonyy/AI-basketball-analysis · src/utils.py:79
y_pos = [(a * (x_val ** 2)) + (b * x_val) + c for x_val in x_pos]
if(shotJudgement == "MISS"):
plt.plot(x, y, 'ro', figure=fig)
plt.plot(x_pos, y_pos, linestyle='-', color='red',
alpha=0.4, linewidth=5, figure=fig)
else:
plt.plot(x, y, 'go', figure=fig)
plt.plot(x_pos, y_pos, linestyle='-', color='green',
alpha=0.4, linewidth=5, figure=fig)
def distance(x, y):
return ((y[0] - x[0]) ** 2 + (y[1] - x[1]) ** 2) ** (1/2)
def calculateAngle(a, b, c):
ba = a - b
bc = c - b