Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

จะทำให้เคลื่อนไหว 3D plot_surface ใน Matplotlib ได้อย่างไร


ในการทำให้ 3D plot_surface เคลื่อนไหวใน Matplotlib เราสามารถทำตามขั้นตอนต่อไปนี้:

  • เริ่มต้นตัวแปรสำหรับจำนวนกริดตาข่าย (N) , ความถี่ต่อวินาที (fps) เพื่อเรียกใช้ฟังก์ชันและหมายเลขเฟรม (frn) .
  • สร้าง x , และ z อาร์เรย์สำหรับเส้นโค้ง
  • สร้างฟังก์ชันเพื่อสร้าง z-array โดยใช้ฟังก์ชันแลมบ์ดา
  • ส่งฟังก์ชันไปยัง คลาสแอนิเมชั่น สร้างฟังก์ชันที่ผู้ใช้กำหนดเองซึ่งลบพล็อตก่อนหน้าและพล็อตพื้นผิวโดยใช้ x , และ z-array
  • สร้างตัวเลขใหม่หรือเปิดใช้งานตัวเลขที่มีอยู่
  • เพิ่มการจัดเรียงแผนย่อยโดยใช้ แผนย่อย() วิธีการ
  • ตั้งค่าขีดจำกัดแกน Z โดยใช้ set_zlim() วิธีการ
  • เรียก คลาสแอนิเมชั่น เพื่อทำให้พื้นผิวเคลื่อนไหว
  • หากต้องการแสดงรูป ให้ใช้ show() วิธีการ

ตัวอย่าง

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()

ผลลัพธ์

จะทำให้เคลื่อนไหว 3D plot_surface ใน Matplotlib ได้อย่างไร