python - 有没有一种简单的方法可以在 matplotlib 中为滚动垂直线设置动画？

Question

我想拥有一个在音频播放实用程序中似乎很常见的进度标记。我认为在 matplotlib 中这相当于左/右动画plt.vlines。我的代码需要 2 秒的数据数组并创建音频时间序列可视化。我正在努力创建一条动画垂直线，该垂直线将从 0 到 2 线性移动 2 秒。

import seaborn as sns
import numpy as np
import matplotlib.pyplot as plt

font = {'weight': 'bold', 'size': 15}
plt.rc('font',**font)
sns.set_style("darkgrid")

testSeries = np.random.randint(-10, 20, 12000)
testSeries = testSeries - testSeries.mean()


fig,axis = plt.subplots(nrows=1,ncols=1,figsize=(18,5),sharex=True)
sns.lineplot(range(0,len(testSeries)),testSeries,  color='#007294')
plt.xlim(0, len(testSeries))
axis.set_xlabel("Time (s)", fontsize='large', fontweight='bold')
axis.set_ylabel("Amplitude", fontsize='large', fontweight='bold')
axis.set_xticklabels(['0', '0.3', '0.6', '1', '1.3', '1.6', '2'],fontsize=15)
fig.tight_layout(rect=[0,0,.8,1]) 
plt.subplots_adjust(bottom=-0.01)
sns.despine()
plt.show()

标签： pythonmatplotlibanimationseaborn

Answer 1

axvline()只需返回一个Line2D对象，因此您可以使用更新其位置Line2D.set_xdata()

duration = 2 # in sec
refreshPeriod = 100 # in ms

fig,ax = plt.subplots()
vl = ax.axvline(0, ls='-', color='r', lw=1, zorder=10)
ax.set_xlim(0,duration)

def animate(i,vl,period):
    t = i*period / 1000
    vl.set_xdata([t,t])
    return vl,

ani = animation.FuncAnimation(fig, animate, frames=int(duration/(refreshPeriod/1000)), fargs=(vl,refreshPeriod), interval=refreshPeriod)
plt.show()

请注意，刷新率并不能保证，它取决于重绘图形所需的时间。你可能不得不玩弄refreshPeriod.