CODEWITHSUNDEEP
pythonmatplotlibnormalize
Rhecsu
Rhecsu

Reputation: 119

Quiver matplotlib : arrow with the same sizes

I'm trying to do a plot with quiver but I would like the arrows to all have the same size.

I use the following input :

q = ax0.quiver(x, y, dx, dy, units='xy' ,scale=1)

But even if add options like norm = 'true' or Normalize = 'true' the arrows are not normalized

Any one knows how to do this ? Thank you

Upvotes: 6

Views: 2672

Answers (1)

Derek Eden
Derek Eden

Reputation: 4618

Not sure if there is a way to do this by explicitly providing a kwarg to plt.quiver, but a quick work around would be like this:

original plot

x = np.arange(10)
y = np.arange(10)
xx, yy = np.meshgrid(x,y)
u = np.random.uniform(low=-1, high=1, size=(10,10))
v = np.random.uniform(low=-1, high=1, size=(10,10))
plt.quiver(xx, yy, u, v)

original

normalized plot

r = np.power(np.add(np.power(u,2), np.power(v,2)),0.5) #could do (u**2+v**2)**0.5, other ways...
plt.quiver(xx, yy, u/r, v/r)

enter image description here

this just normalizes the u and v components of the vector by the magnitude of the vector, thereby maintaining the proper direction, but scaling down the magnitude of each arrow to 1

Upvotes: 7

Related Questions