如何在while循环中更新matplotlib图？

Question

我正在尝试编写一个绘制图形的脚本，并要求用户选择两个点。然后绘制一个矩形，其中的点是左下角和右上角。然后重复该过程，每次迭代添加一个新矩形。但是，我希望它在选择当前点后立即绘制矩形，但在选择下一组点之前不会发生。任何人都知道如何解决这个问题？

import matplotlib.pylab as plt

x = rand(10,10)
plt.imshow(x)

select = 'y'
while select == 'y':

    select = (raw_input('select points (y or n) ? '))
    a, b = plt.ginput(2)
    plt.vlines(a[0], a[1], b[1], lw=5)
    plt.vlines(b[0], a[1], b[1], lw=5)
    plt.hlines(a[1], a[0], b[0], lw=5)
    plt.hlines(b[1], a[0], b[0], lw=5)
    plt.draw()