How to pause a pylab figure until a key is pressed or mouse is clicked? (Matplotlib)

To pause a pylab figure until a key is pressed of mouse is clicked, we can use"button_press_event" key event.

Steps

• Set the figure size and adjust the padding between and around the subplots.
• Set the "TkAgg" background.
• Turn the interactive mode ON.
• Create a new figure or activate an existing figure.
• Make a variable, pause=False.
• Whenever "button_press_event", pause the figure.
• Bind the function to the event.
• Create data, x and y data points using numpy.
• Iterate a True loop to change the plot line and color.
• To display the figure, use show() method.

Example

import matplotlib from matplotlib import pyplot as plt import numpy as np plt.rcParams["figure.figsize"] = [7.50, 3.50] plt.rcParams["figure.autolayout"] = True matplotlib.use("TkAgg") plt.ion() fig = plt.figure() pause = False def onclick(event):    global pause    pause = not pause fig.canvas.mpl_connect('button_press_event', onclick) data = np.linspace(-10, 10, 100) x = np.sin(data) y = np.cos(data) flag = 1 while True:    if not pause:       if flag == 1:          fig.clear()          plt.plot(data, y, color='red')          flag = 0     else:       fig.clear()       plt.plot(data, x, color='yellow')       flag = 1 fig.canvas.get_tk_widget().update()

Output

You will see sine and cosine waves alternatively in yellow and red until you click a mouse button. The figure will pause after clicking either of the mouse buttons.