Интерактивный график matplotlib с двумя ползунками

Я библиотек matplotlib для создания некоторого графика, который зависит от 8 переменных. Я хотел бы изучить, как меняется сюжет, когда я меняю некоторые из них. Я создал сценарий, который вызывает библиотек matplotlib один и генерирует разные снимки, которые позже я конвертирую в фильм, это неплохо, но немного неуклюже.

  1. интересно, могу ли я как-то взаимодействовать с регенерацией сюжета с помощью клавиш клавиатуры для увеличения / уменьшения значений некоторых из переменные и мгновенно увидеть, как меняется сюжет.

  2. каков наилучший подход для этого?

  3. также, если вы можете указать мне на интересные ссылки или ссылку с примером сюжета только с двумя ползунками?

4 ответов


в дополнение к тому, что @triplepoint упоминалось, посмотрите на виджет слайдера.

здесь пример на странице примеров matplotlib. Это графический ползунок, а не привязки клавиатуры, но он работает довольно хорошо для того, что вы хотите сделать.

также обратите внимание, что, чтобы гарантировать, что ползунки и кнопки остаются отзывчивыми, а не собранными мусором, ссылки на объекты (amp_slider, freq_slider, etc.) должно поддерживаться себе.

(я делаю это сообщество wiki, так как я просто копирую вставку из примера. этот конкретный пример учит плохим привычкам (например,from pylab import *), но он получает точку в поперечнике. пример был исправлен, чтобы избежать использования pylab.)

from numpy import pi, sin
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.widgets import Slider, Button, RadioButtons

def signal(amp, freq):
    return amp * sin(2 * pi * freq * t)

axis_color = 'lightgoldenrodyellow'

fig = plt.figure()
ax = fig.add_subplot(111)

# Adjust the subplots region to leave some space for the sliders and buttons
fig.subplots_adjust(left=0.25, bottom=0.25)

t = np.arange(0.0, 1.0, 0.001)
amp_0 = 5
freq_0 = 3

# Draw the initial plot
# The 'line' variable is used for modifying the line later
[line] = ax.plot(t, signal(amp_0, freq_0), linewidth=2, color='red')
ax.set_xlim([0, 1])
ax.set_ylim([-10, 10])

# Add two sliders for tweaking the parameters

# Define an axes area and draw a slider in it
amp_slider_ax  = fig.add_axes([0.25, 0.15, 0.65, 0.03], axisbg=axis_color)
amp_slider = Slider(amp_slider_ax, 'Amp', 0.1, 10.0, valinit=amp_0)

# Draw another slider
freq_slider_ax = fig.add_axes([0.25, 0.1, 0.65, 0.03], axisbg=axis_color)
freq_slider = Slider(freq_slider_ax, 'Freq', 0.1, 30.0, valinit=freq_0)

# Define an action for modifying the line when any slider's value changes
def sliders_on_changed(val):
    line.set_ydata(signal(amp_slider.val, freq_slider.val))
    fig.canvas.draw_idle()
amp_slider.on_changed(sliders_on_changed)
freq_slider.on_changed(sliders_on_changed)

# Add a button for resetting the parameters
reset_button_ax = fig.add_axes([0.8, 0.025, 0.1, 0.04])
reset_button = Button(reset_button_ax, 'Reset', color=axis_color, hovercolor='0.975')
def reset_button_on_clicked(mouse_event):
    freq_slider.reset()
    amp_slider.reset()
reset_button.on_clicked(reset_button_on_clicked)

# Add a set of radio buttons for changing color
color_radios_ax = fig.add_axes([0.025, 0.5, 0.15, 0.15], axisbg=axis_color)
color_radios = RadioButtons(color_radios_ax, ('red', 'blue', 'green'), active=0)
def color_radios_on_clicked(label):
    line.set_color(label)
    fig.canvas.draw_idle()
color_radios.on_clicked(color_radios_on_clicked)

plt.show()

Example


Matplotlib имеет некоторые довольно приятные функции gui. Есть несколько примеров документации в исходном tarball matplotlib, в /examples/user_interfaces и matplotlib>/examples/event_handling. Конкретно на keyhandling:http://matplotlib.sourceforge.net/examples/event_handling/keypress_demo.html

Я сделал что-то похожее на то, что вы стремитесь:

import numpy as np
import pylab

class plotter:
    def __init__(self, initial_values):
        self.values
        self.fig = pylab.figure()
        pylab.gray()
        self.ax = self.fig.add_subplot(111)
        self.draw()
        self.fig.canvas.mpl_connect('key_press_event',self.key)

    def draw(self):
        im = your_function(self.values)
        pylab.show()
        self.ax.imshow(im)

    def key(self, event):
        if event.key=='right':
            self.values = modify()
        elif event.key == 'left':
            self.values = modify()

        self.draw()
        self.fig.canvas.draw()

я использовал это для переключения отображения различных изображения в стеке на нажатиях клавиш, но вы должны иметь возможность вводить логику для изменения значений при вводе с клавиатуры.

Если вы хотите делать такие вещи, как иметь значения ввода пользователя, я думаю, что в примерах есть опции для диалоговых окон, но если вы просто хотите увеличить/уменьшить количество переменных, просто определение пар клавиатуры для них таким образом может работать хорошо


использовать waitforbuttonpress(timeout=0.001) тогда сюжет будет видеть ваши тики мыши.


Я не думаю, что просто построение графиков, используя plt.plot позволит вам сделать это. Вам нужно будет самостоятельно сделать пользовательский скрипт/приложение GUI, встраивая в него Matplotlib. В настоящее время Matplotlib поддерживает все основные наборы инструментов GUI - PyGTK+, PyQt4 и wxPython.
Я использую wxPython и встраивание matplotlib в него довольно легко. Аналогично должно быть дело с другими GUI toolkits. Вы можете получить всю необходимую для этого информацию в книге - enter image description here

это доступно на amazon здесь.