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

Vẽ đồ thị rung động trong Python bằng Matplotlib

Để tạo hiệu ứng rung trong Python, chúng ta có thể thực hiện các bước sau -

  • Đặ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 x y điểm dữ liệu bằng cách sử dụng numpy.
  • Tạo u v điểm dữ liệu bằng cách sử dụng numpy.
  • Tạo một hình và một tập hợp các tình huống phụ.
  • Vẽ một trường mũi tên 2D bằng cách sử dụng quiver () phương pháp.
  • Để tạo hoạt ảnh cho bộ rung, chúng tôi có thể thay đổi u v giá trị, trong animate () phương pháp. Cập nhật u v giá trị và màu của vectơ.
  • Để hiển thị hình này, hãy sử dụng show () phương pháp.

Ví dụ

import numpy as np
import random as rd
from matplotlib import pyplot as plt, animation

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

x, y = np.mgrid[:2 * np.pi:10j, :2 * np.pi:5j]
u = np.cos(x)
v = np.sin(y)

fig, ax = plt.subplots(1, 1)
qr = ax.quiver(x, y, u, v, color='red')

def animate(num, qr, x, y):
   u = np.cos(x + num * 0.1)
   v = np.sin(y + num * 0.1)
   qr.set_UVC(u, v)
   qr.set_color((rd.random(), rd.random(), rd.random(), rd.random()))
   return qr,

anim = animation.FuncAnimation(fig, animate, fargs=(qr, x, y),
                              interval=50, blit=False)

plt.show()

Đầu ra

Vẽ đồ thị rung động trong Python bằng Matplotlib