Plotting Vector Fields in Python · Ajit Kumar
| Created | |
|---|---|
| Tags | |
| URL | https://krajit.github.io/sympy/vectorFields/vectorFields.html |
Vector Fields
Example: Plot of vector field F⃗ (x,y)=i−j
F⃗ (x,y)=i−j
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
x,y = np.meshgrid(np.linspace(-5,5,10),np.linspace(-5,5,10))
u = 1
v = -1
plt.quiver(x,y,u,v)
plt.show()

Example: Plot of vector field F⃗ (x,y)=xx2+y2√i+yx2+y2√jF→(x,y)=xx2+y2i+yx2+y2j
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
x,y = np.meshgrid(np.linspace(-5,5,10),np.linspace(-5,5,10))
u = x/np.sqrt(x**2 + y**2)
v = y/np.sqrt(x**2 + y**2)
plt.quiver(x,y,u,v)
plt.show()

Example: Plot of vector field F⃗ (x,y)=−yx2+y2√i+xx2+y2√jF→(x,y)=−yx2+y2i+xx2+y2j
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline
x,y = np.meshgrid(np.linspace(-5,5,10),np.linspace(-5,5,10))
u = -y/np.sqrt(x**2 + y**2)
v = x/np.sqrt(x**2 + y**2)
plt.quiver(x,y,u,v)
plt.show()

plt.quiverkey?
3d Vector Field
%reset -f
from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
import numpy as np
%matplotlib inline
fig = plt.figure()
ax = fig.gca(projection='3d')
x, y, z = np.meshgrid(np.arange(-0.8, 1, 0.2),
np.arange(-0.8, 1, 0.2),
np.arange(-0.8, 1, 0.8))
u = np.sin(np.pi * x) * np.cos(np.pi * y) * np.cos(np.pi * z)
v = -np.cos(np.pi * x) * np.sin(np.pi * y) * np.cos(np.pi * z)
w = (np.sqrt(2.0 / 3.0) * np.cos(np.pi * x) * np.cos(np.pi * y) *
np.sin(np.pi * z))
ax.quiver(x, y, z, u, v, w, length=0.1, color = 'black')
plt.show()
