Putting arrowheads on vectors in Matplotlib's 3D plot

Matplotlib Server Side Programming Programming

To draw arrow heads vectors in 3D matplotlb's plot, we can take the following steps −

Create a 2D array, where x, y, z, u, v and w are the coordinates of the arrow locations and direction components of arrow vectors.
Using figure() method, create a new figure or activate an existing figure.
Add an '~.axes.Axes' to the figure as part of a subplot arrangement, using add_subplot() method.
Plot a 3D field of arrows, using quiver() method.
Using ylim, xlim, zlim, limit the range of the axes.
Set the title of the plot.
To display the figure, use show() method.

Example

import matplotlib.pyplot as plt
import numpy as np
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
soa = np.array([[0, 0, 1, 1, -2, 0], [0, 0, 2, 1, 1, 0], [0, 0, 3, 2, 1, 0], [0, 0, 4, 0.5, 0.7, 0]])
X, Y, Z, U, V, W = zip(*soa)
fig = plt.figure()
ax = fig.add_subplot(projection='3d')
ax.quiver(X, Y, Z, U, V, W, color='red')
ax.set_xlim([-1, 0.5])
ax.set_ylim([-1, 1.5])
ax.set_zlim([-1, 8])
ax.set_title("Vectors")
plt.show()

Output

Rishikesh Kumar Rishi

Published on 07-May-2021 07:46:41

Previous Page Print Page