Python анимация полярный сюжет - PullRequest
Новая
       12

Python анимация полярный сюжет

0 голосов
/ 07 июня 2018

Я пытался создать анимацию с этим сюжетом, где с увеличением «бета» кривые растягиваются, но я не смог.Пожалуйста, кто-нибудь может мне помочь? 

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

# Data:
mu = 1
e = 1
a = 1
c = 1
beta = np.linspace(0.0 ,0.5,25)
C = (mu*((e*a)**2)) / (16*(np.pi**2)*c) 

theta = np.linspace(-2 * np.pi, 2 * np.pi, 200)

fig = plt.figure(figsize=(6,6))
ax = plt.subplot(111, polar=True)

for i in range(len(beta)):
    b = beta[i]
    r = ((np.sin(theta))**2) / ((1 - b*np.cos(theta))**5)
    dP = C*r
    ax.plot(theta, dP)
    ax.set_yticklabels([])
    if i==10000:
        break

plt.show()

1 Ответ

0 голосов
/ 07 июня 2018

Вы должны создать функцию, которая отвечает за обновление вашего графика, в этом случае я назову его update(), как показано ниже: 

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

# Data:
mu = 1
e = 1
a = 1
c = 1

beta = np.linspace(0.0 ,0.5,25)
C = (mu*((e*a)**2)) / (16*(np.pi**2)*c) 

theta = np.linspace(-2 * np.pi, 2 * np.pi, 200)

fig = plt.figure(figsize=(6,6))
ax = plt.subplot(111, polar=True)

line, = ax.plot([],[])


def update(b):
    r = (np.sin(theta)**2)/(1 - b*np.cos(theta))**5
    dP = C*r
    line.set_xdata(theta)
    line.set_ydata(dP)
    return line,

ani = FuncAnimation(fig, update, frames=beta, blit=True)
plt.show()
...