def plot_vector_direction(coordAmp):
"""Plots average vector direction as color"""
tails = trackManipulations.find_track_starts(coordAmp)
heads = trackManipulations.find_average_positions(coordAmp)
tailArray = np.array(tails)
headArray = np.array(heads)
vectors = list(headArray - tailArray)
x,y = zip(*tails)
u,v = zip(*vectors)
directions = trackManipulations.generate_directions(vectors)
scaledDirections = np.array(directions)/(2*pi)
plt.clf()
plt.figure()
plt.scatter(x,y,c=scaledDirections)
plt.draw()
plt.show()
def plot_mean_vectors(coordAmp):
"""Plots the vectors of average motion"""
tails = trackManipulations.find_track_starts(coordAmp)
heads = trackManipulations.find_average_positions(coordAmp)
tailArray = np.array(tails)
headArray = np.array(heads)
vectors = list(headArray - tailArray)
x,y = zip(*tails)
u,v = zip(*vectors)
plt.clf()
plt.figure()
ax = plt.gca()
ax.set_aspect('equal', adjustable='box')
ax.quiver(x,y,u,v,angles='xy', scale_units='xy', scale=1)
# ax.set_xlim([60,80])
# ax.set_ylim([180,200])
# plt.draw()
plt.show()
