Question

我想暂停一个我在自己的事件循环中运行的动画。以下是代码的简化版本：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.widgets import Button
from mpl_toolkits.mplot3d import proj3d

import time

def main():
    global paused
    paused = False

    fig = plt.figure()
    ax = fig.add_subplot(111, projection='3d')

    ax.set_ylim(-100, 100)
    ax.set_xlim(-10, 10)
    ax.set_zlim(-100, 100)

    plt.ion()
    plt.show()

    def pause_anim(event):
        global paused
        paused = not paused

    pause_ax = fig.add_axes((0.7, 0.03, 0.1, 0.04))
    pause_button = Button(pause_ax, 'pause', hovercolor='0.975')
    pause_button.on_clicked(pause_anim)

    x = np.arange(-50, 51)

    line = ax.plot([], [], [], c="r")[0]

    y_range = list(np.arange(1, 60, 3))
    y_len = len(y_range)
    idx = 0

    while True:

        if not paused:
            idx += 1
            if idx >= y_len:
                break

            y = y_range[idx]
            z = - x**2 + y - 100

            line.set_data(x, 0)
            line.set_3d_properties(z)

            plt.draw()
            plt.pause(0.2)

        else:
            time.sleep(1)  # this stops button events from happening
            #input("Shoop?")  # prompting for input works
            # I've also tried putting a mutex here

if __name__ == '__main__':
    main()

正如我在代码中提到的那样，我已尝试过time.sleep和Lock，但是一旦我暂停，这些就会阻止我暂停。如何在不破坏恢复动画的能力的情况下暂停循环？

Answer 1

您可以将time.sleep(1) - 部分中的else替换为plt.pause，就像您在if - 部分中所做的那样。

        ....
        plt.draw()
        plt.pause(0.2)
    else:
        plt.pause(0.2)

另一个选项当然是使用matplotlib动画，例如FuncAnimation，其方法为.event_source.stop()和.event_source.start()。（如本问题的两个答案所示：stop / start / pause in python matplotlib animation）

暂停matplotlib自定义动画循环

1 个答案: