Vector field - Quiver plot¶
In [1]:
import numpy as np
import matplotlib.pyplot as plt
x_range = np.linspace(-3, 3, 21)
y_range = np.linspace(-3, 3, 21)
(x, y) = np.meshgrid(x_range, y_range)
u = np.sin(y)
v = np.cos(x)
mag = np.sqrt(u**2+v**2)
plt.style.use('dark_background')
fig, ax = plt.subplots()
q = plt.quiver(x, y, u, v, mag, pivot='mid', cmap='plasma')
cb = plt.colorbar(q)
cb.set_label('Magnitude')
plt.show()
