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明确绘制输入数据点之间的步骤。它永远不会绘制部分＆＃34;步骤＆＃34;。

您想要使用＆＃34;部分步骤＆＃34;介于两者之间。

不是使用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()

注意部分＆＃34;步骤＆＃34;在开始和结束

将此结合到动画示例中，我们会得到类似的内容：

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