Cách animate 3D plot_surface trong Matplotlib?

Để animate 3D plot_surface trong Matplotlib, chúng ta có thể thực hiện các bước sau:

• Khởi tạo biến cho số lưới mesh (N), tần suất trên giây (fps) gọi hàm, và số frame (frn).
• Tạo mảng x, y và z cho đường cong.
• Tạo hàm làm z-array dùng lambda function.
• Để truyền hàm vào animation class, tạo hàm user-defined xóa plot trước và vẽ surface dùng x, y và z-array.
• Tạo figure mới hoặc kích hoạt figure hiện có.
• Thêm arrangement subplot dùng subplots().
• Đặt giới hạn trục Z dùng set_zlim().
• Gọi animation class để animate surface plot.
• Để hiển thị figure, dùng phương thức show().

Ví dụ

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation

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

N = 50
fps = 250
frn = 75

x = np.linspace(-4, 4, N + 1)
x, y = np.meshgrid(x, x)

zarray = np.zeros((N + 1, N + 1, frn))

f = lambda x, y, sig: 1 / np.sqrt(sig) * np.exp(-(x ** 2 + y ** 2) / sig ** 2)

for i in range(frn):
    zarray[:, :, i] = f(x, y, 1.5 + np.sin(i * 2 * np.pi / frn))

def change_plot(frame_number, zarray, plot):
    plot[0].remove()
    plot[0] = ax.plot_surface(x, y, zarray[:, :, frame_number], cmap="afmhot_r")

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
plot = [ax.plot_surface(x, y, zarray[:, :, 0], color='0.75', rstride=1, cstride=1)]
ax.set_zlim(0, 1.1)

ani = animation.FuncAnimation(fig, change_plot, frn, fargs=(zarray, plot), interval=1000 / fps)
ax.axis('off')

plt.show()

Kết quả