Matplotlib анимация шага

Question

Я создаю анимацию Matplotlib функции шага. Я использую следующий код ...

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

fig = plt.figure() 
ax = plt.axes(xlim=(0, 2), ylim=(-2, 2)) 

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

def init(): 
    line.set_data([], []) 
    return line, 

def animate(i): 
    x = np.linspace(0, 2, 10) 
    y = np.sin(2 * np.pi * (x - 0.01 * i)) 
    line.set_data(x, y) 
    return line, 

anim = animation.FuncAnimation(fig, animate, init_func=init, 
           frames=100, interval=20, blit=True) 

plt.show() 

Он смутно напоминает то, что я хочу (что-то вроде ниже GIF), но вместо значений постоянен и прокрутка с течением времени каждый шаг является динамичным и сдвигает вверх и вниз. Как бы изменился мой код, чтобы добиться этого изменения?

Answer 1

step явно участки шагов между точками ввода данных. Он никогда не может нарисовать частичный «шаг».

Вы хотите анимацию с «частичными шагами» между ними.

Вместо того, чтобы использовать ax.step, используйте ax.plot, но сделайте ступенчатую серию, построив y = y - y % step_size.

Другими словами, что-то вроде:

import numpy as np 
import matplotlib.pyplot as plt 

x = np.linspace(0, 10, 1000) # Using a series of 1000 points... 
y = np.sin(x) 

# Make *y* increment in steps of 0.3 
y -= y % 0.3 

fig, ax = plt.subplots() 
ax.plot(x, y) 
plt.show() 

Обратите внимание, частичные «шаги» в начале и в конце

Включение этого в вашем примере анимации, мы бы получить что-то похожее на:

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

fig = plt.figure() 
ax = plt.axes(xlim=(0, 2), ylim=(-2, 2)) 

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

def init(): 
    line.set_data([], []) 
    return line, 

def animate(i): 
    x = np.linspace(0, 2, 1000) 
    y = np.sin(2 * np.pi * (x - 0.01 * i)) 
    y -= y % 0.3 
    line.set_data(x, y) 
    return line, 

anim = animation.FuncAnimation(fig, animate, init_func=init, 
           frames=100, interval=20, blit=True) 

plt.show()