Matplotlib - Checkbox Widget

Matplotlib - Checkbox Widget

In Matplotlib, you can create interactive plots with various widgets, including checkboxes. The matplotlib.widgets.CheckButtons class provides functionality for adding checkboxes to your plots. These checkboxes can be used to toggle the visibility of data, change styles, or trigger other custom actions.

Here's a basic example to illustrate how to create and use checkboxes in Matplotlib:

Example Code

import matplotlib.pyplot as plt from matplotlib.widgets import CheckButtons # Sample data t = range(10) y1 = [2*i for i in t] y2 = [3*i for i in t] # Create the figure and the line that we will manipulate fig, ax = plt.subplots() line1, = ax.plot(t, y1, label='Line 1') line2, = ax.plot(t, y2, label='Line 2') # Add a check button widget rax = plt.axes([0.05, 0.5, 0.1, 0.15]) # x, y, width, height check = CheckButtons(rax, ('Line 1', 'Line 2'), (True, True)) # Define the functionality when checkbox is clicked def func(label): if label == 'Line 1': line1.set_visible(not line1.get_visible()) elif label == 'Line 2': line2.set_visible(not line2.get_visible()) plt.draw() # Attach the callback function to the check buttons check.on_clicked(func) plt.show()

Explanation

  • Creating the Plot: Two lines (line1 and line2) are plotted on the axes ax.
  • Adding CheckButtons: A CheckButtons widget is created with two checkboxes corresponding to the two lines. The position and size of the checkbox area are set by [0.05, 0.5, 0.1, 0.15] (x, y, width, height).
  • Callback Function: The func function is defined to toggle the visibility of line1 and line2 based on the state of the checkboxes.
  • Registering the Callback: check.on_clicked(func) connects the func function to the checkbox click event.

When you run this code, a Matplotlib window with the plot and checkboxes will appear. Clicking on the checkboxes will toggle the visibility of the respective lines in the plot. This example is a simple demonstration, but you can extend this approach for more complex interactions in your plots.

More Tags

bootstrap-modal google-forms enumerator variable-names decimalformat concurrent-collections azure-servicebus-topics tortoisegit react-server sequential

More Programming Guides

Other Guides

More Programming Examples