使用matplotlib滑块更新轮廓图水平

时间:2015-01-06 13:17:49

标签: python user-interface matplotlib widget

我尝试使用滑块更改matplotlib填充轮廓图上的颜色级别值。即contourf(x,y,z,np.linspace(a,b,n)),滑块将控制a和b,并在移动滑块时更改绘图颜色级别。 以下代码采用列格式化数据将其转换为contourf所需的格式,然后实现滑块。 这就是我尝试过的:

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.widgets import Slider

data=np.genfromtxt('file.dat',skip_header=1)
len=np.sqrt(data[:,0].size)
x=np.reshape(data[:,0],(len,len))       
y=np.reshape(data[:,1],(len,len))
z=np.reshape(data[:,3],(len,len))

l=plt.contourf(x,y,z,np.linspace(0,100,255))

axmax = plt.axes([0.25, 0.1, 0.65, 0.03])  #slider location and size
axmin  = plt.axes([0.25, 0.15, 0.65, 0.03])
smax = Slider(axmax, 'Max',0, 100, 50)      #slider properties
smin = Slider(axmin, 'Min', 0, 100, 0)

def update(val):
    l.levels(np.linspace(smin.val,smax.val,255))#changing levels of plot
    fig.canvas.draw_idle()                      #line that throws error
smax.on_changed(update)
smin.on_changed(update)

plt.show()

移动滑块时会抛出大量的matplotlib错误,其中相关的滑块是' TypeError:' numpy.ndarray'对象不可调用'这是由行抛出的

fig.canvas.draw_idle()

1 个答案:

答案 0 :(得分:2)

问题是l.levels是一个数组,因此您必须更改此数组中的值。在我的测试中,更改这些值不会导致绘图更新。所以另一种解决方案是清除轴并重绘图。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.widgets import Slider

data=np.random.random([25,4])
data = data*100
len=np.sqrt(data[:,0].size)
x=np.reshape(data[:,0],(len,len))       
y=np.reshape(data[:,1],(len,len))
z=np.reshape(data[:,3],(len,len))

l=plt.contourf(x,y,z,np.linspace(0,100,255))
contour_axis = plt.gca()

axmax = plt.axes([0.25, 0.1, 0.65, 0.03])  #slider location and size
axmin  = plt.axes([0.25, 0.15, 0.65, 0.03])
smax = Slider(axmax, 'Max',0, 100, 50)      #slider properties
smin = Slider(axmin, 'Min', 0, 100, 0)


def update(val):
    contour_axis.clear()
    contour_axis.contourf(x,y,z,np.linspace(smin.val,smax.val,255))
    plt.draw()                   
smax.on_changed(update)
smin.on_changed(update)

plt.show()