Computer >> Máy Tính >  >> Lập trình >> Python

Tính độ cong của một trường vectơ bằng Python và vẽ biểu đồ đó bằng Matplotlib

Để tính độ cong của trường vectơ trong Python và vẽ biểu đồ bằng Matplotlib, chúng ta có thể sử dụng quiver () và tính toán dữ liệu tương ứng.

Các bước

  • Đặt kích thước hình và điều chỉnh phần đệm giữa và xung quanh các ô phụ.
  • Tạo một hình mới hoặc kích hoạt một hình hiện có bằng cách sử dụng figure () phương pháp.
  • Thêm các trục 3D vào hình như một phần của sắp xếp ô con.
  • Tạo x , y z điểm dữ liệu bằng cách sử dụng lưới phức hợp.
  • Tạo u , v w vị trí vectơ cuộn dữ liệu.
  • Sử dụng quiver () phương pháp lấy vectơ.
  • Tắt các trục.
  • Để hiển thị hình này, hãy sử dụng show () phương pháp.

Ví dụ

import matplotlib.pyplot as plt
import numpy as np

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

fig = plt.figure()

ax = fig.add_subplot(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 = 0
v = y**2
w = -2*y*z - y

ax.quiver(x, y, z, u, v, w, length=0.1)
ax.axis('off')

plt.show()

Đầu ra

Tính độ cong của một trường vectơ bằng Python và vẽ biểu đồ đó bằng Matplotlib