Matplotlib: контурный график с виджетом слайдера

Question

Matplotlib: контурный график с виджетом слайдера

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

AttributeError: QuadContourSet instance has no attribute 'set_data'

Я подозреваю, что вызываю set_data на неправильном объекте, но я не могу найти никакой документации о том, что такоеправильный объект. Вы можете помочь? Спасибо.

Вот полный код:

import numpy as np

import matplotlib as mpl

import matplotlib.mlab as mlab

import matplotlib.pyplot as pyl

from matplotlib.contour import QuadContourSet

from matplotlib.widgets import Slider



#Define display parameters

mpl.rcParams['xtick.direction'] = 'out'

mpl.rcParams['ytick.direction'] = 'out'

delta = 0.025



#Define model parameters

alpha = .5

beta = .5

x_bar, a, b, c = 2, 0, 1, .1

v = np.arange(0, 10, delta)

w = np.arange(0, 10, delta)



#Calculate grid values

V, W = np.meshgrid(v,w)

Z = (V**(beta))*(W**(1-beta))

X = x_bar + a + b*Z

U = alpha*np.log(V) + (1-alpha)*np.log(X) - c*(W+V)



# Plot

fig = pyl.figure()



ax = fig.add_subplot(221)

CS = QuadContourSet(pyl.gca(), V, W, U, 200)

pyl.clabel(CS, inline=1, fontsize=10)

pyl.title('Simplest default with labels')



#Define slider for alpha

axcolor = 'lightgoldenrodyellow'

alpha_axis  = pyl.axes([0.25, 0.15, 0.65, 0.03], axisbg=axcolor)

alpha_slider = Slider(alpha_axis, 'Amp', 0, 1, valinit=.5)



def update(val):

    alpha = alpha_slider.val

    U = alpha*np.log(V) + (1-alpha)*np.log(X) - c*(W+V)

    CS.set_data(V, W, U)

    pyl.draw()



alpha_slider.on_changed(update)



pyl.show()

523 1

python matplotlib slider

1 ответ:

Comments

Ничего не найдено.

Jouni K. Seppänen · Accepted Answer · 2012-02-08 22:28:08

Проблема заключается в том, что объект QuadContourSet не может обновить свои данные, так как если вы произвольно измените данные, то все это должно быть пересчитано. Я не знаю, есть ли что-то в вашем конкретном способе генерирования данных, что позволило бы проще модифицировать контурные линии, но если нет, я думаю, что вам нужно сделать, это построить контуры с нуля:
# After your "Define model parameters" block

def compute_and_plot(ax, alpha):
    #Calculate grid values
    V, W = np.meshgrid(v,w)
    Z = (V**(beta))*(W**(1-beta))
    X = x_bar + a + b*Z
    U = alpha*np.log(V) + (1-alpha)*np.log(X) - c*(W+V)

    CS = QuadContourSet(ax, V, W, U, 200)
    pyl.clabel(CS, inline=1, fontsize=10)

# Plot
fig = pyl.figure()
pyl.title('Simplest default with labels')
ax = fig.add_subplot(221)
compute_and_plot(ax, alpha)

#Define slider for alpha
axcolor = 'lightgoldenrodyellow'
alpha_axis  = pyl.axes([0.25, 0.15, 0.65, 0.03], axisbg=axcolor)
alpha_slider = Slider(alpha_axis, 'Amp', 0, 1, valinit=.5)

def update(ax, val):
    alpha = alpha_slider.val
    ax.cla()
    compute_and_plot(ax, alpha)
    pyl.draw()

alpha_slider.on_changed(lambda val: update(ax, val))

pyl.show()