Vẽ trường vector (vector field) trên trục tọa độ trong Python Matplotlib

Để vẽ trường vector trên trục tọa độ trong matplotlib, ta thực hiện:

• Thiết lập kích thước hình và điều chỉnh khoảng đệm giữa và xung quanh subplots.
• Tạo X, Y, T, R, U, V dùng numpy.
• Thêm trục vào hình hiện tại và làm trục hiện tại.
• Vẽ trường mũi tên 3D dùng quiver().
• Hiển thị hình dùng show().

Ví dụ

import numpy as np
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

n = 8
X, Y = np.mgrid[0:n, 0:n]
T = np.arctan2(Y - n / 2., X - n/2.)
R = 10 + np.sqrt((Y - n / 2.0) ** 2 + (X - n / 2.0) ** 2)
U, V = R * np.cos(T), R * np.sin(T)
plt.axes([0.025, 0.025, 0.95, 0.95])
plt.quiver(X, Y, U, V, R, cmap="copper")
plt.show()

Kết quả