matplotlib에서 회전하는 3D 그래프를 만들려면 애니메이션 을 사용할 수 있습니다. 함수를 반복적으로 호출하기 위한 클래스입니다.
단계
-
메쉬 그리드 수, 함수를 호출하는 초당 빈도, 프레임 수에 대한 변수를 초기화합니다.
-
곡선에 대한 x, y 및 z 배열을 만듭니다.
-
람다 함수를 사용하여 z 배열을 만드는 함수를 만듭니다.
-
애니메이션 클래스에 함수를 전달하려면 사용자 정의 함수를 만들어 이전 플롯을 제거하고 x, y 및 zarray를 사용하여 표면을 플롯합니다.
-
새 그림을 만들거나 기존 그림을 활성화하세요.
-
subplots() 메서드를 사용하여 subplot 배열을 추가합니다.
-
set_zlim() 을 사용하여 Z축 제한을 설정합니다. 방법.
-
애니메이션 호출 표면 플롯을 애니메이션화하는 클래스입니다.
-
애니메이션 플롯을 표시하려면 show() 를 사용하십시오. 방법.
예시
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
plt.rcParams["figure.figsize"] = [7.00, 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)
ani.save('526.gif')
plt.show() 출력
