如何在matplotlib中以交互方式绘制垂直线？

Question

我有一个时间序列图，我需要画一条移动的垂直线来显示兴趣点。

我使用以下玩具示例来完成相同的操作。但是，它会同时打印所有行，而我想一次显示这些垂直线。

import time
ion() # turn interactive mode on

# initial data
x = arange(-8, 8, 0.1);
y1 = sin(x)
y2 = cos(x)
line1, = plt.plot(x, y1, 'r')
xvals = range(-6, 6, 2);
for i in xvals:
    time.sleep(1)
    # update data
    plt.vlines(i, -1, 1, linestyles = 'solid', color= 'red')
    plt.draw()

Answer 1

如果我理解得很好，你想使用matplotlib的animation工具。一个例子（改编自doc）：

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

X_MIN = -6
X_MAX = 6
Y_MIN = -1
Y_MAX = 1
X_VALS = range(X_MIN, X_MAX+1) # possible x values for the line


def update_line(num, line):
    i = X_VALS[num]
    line.set_data( [i, i], [Y_MIN, Y_MAX])
    return line, 

fig = plt.figure()

x = np.arange(X_MIN, X_MAX, 0.1);
y = np.sin(x)

plt.scatter(x, y)

l , v = plt.plot(-6, -1, 6, 1, linewidth=2, color= 'red')

plt.xlim(X_MIN, X_MAX)
plt.ylim(Y_MIN, Y_MAX)
plt.xlabel('x')
plt.ylabel('y = sin(x)')
plt.title('Line animation')

line_anim = animation.FuncAnimation(fig, update_line, len(X_VALS), fargs=(l, ))

#line_anim.save('line_animation.gif', writer='imagemagick', fps=4);

plt.show()

结果gif如下所示：

Answer 2

你可以尝试在plt.vlines之后调用plt.draw吗？ plt.draw用于在修改后以交互方式重绘图形。