使用滑块选择要显示的图形

问题描述

假设我想通过读取三个.xlsx文件来显示三个模拟。 接下来，我想设计一个滑块来选择要显示的模拟。 如果我将滑块移到0，那么0将是函数"；update()"；的输入。将显示第一个模拟。

代码如下：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import cm
from matplotlib.widgets import Slider
import pandas as pd
import ipywidgets as wg

# input files

rm = pd.read_excel("test_3d.xlsx", header = None)
rm1 = pd.read_excel("test_3d1.xlsx", header = None)
rm2 = pd.read_excel("test_3d2.xlsx", header = None)
rec = np.shape(rm)
X = np.arange(1,rec[1]+1,1)
Y = np.arange(1,rec[0]+1,1)
x , y = np.meshgrid(X,Y)


# Set 3D plots

fig = plt.figure()
ax1 = fig.add_axes([0, 0, 1, 0.8], projection = '3d')


# Choose which 3D plots to show

def update(val):
    if val == 0:
        ax1.cla()
        ax1.plot_surface(x, y, rm, cmap = cm.coolwarm, linewidth = 0, antialiased = False)        
    elif val == 1:
        ax1.cla()
        ax1.plot_surface(x, y, rm1, cmap = cm.coolwarm, linewidth = 0, antialiased = False)
    elif val == 2:
        ax1.cla()
        ax1.plot_surface(x, y, rm2, cmap = cm.coolwarm, linewidth = 0, antialiased = False)      
    
ax1.set_zlim(-110, -80)

# Design a slider to choose which simulation to show 

slider = wg.IntSlider(value=1, min=0, max=2, description='this is slider')
slideroutput = wg.Output()
display(slider, slideroutput)


numberonslider = []
def on_value_change(change):

    with slideroutput:
        numberonslider.append(change['new'])
        print(numberonslider[-1])
    
    
ddd = slider.observe(on_value_change, names='value')
update(ddd)

如果我移动滑块，"；ddd"；会显示0、1或2的列表。

但是，3D模拟不会显示。如何修改代码？

---

解决方案

我正在使用JupyterLab。我需要%matplotlib widget用于任何类型的交互式matplotlib绘图。下面的代码运行正常，但是没有%matplotlib widget就无法运行。

警告：%matplotlib widget与import matplotlib.widget

不同

您没有提供任何示例数据，所以我只是编造了一些数据。基本上您的代码结构不正确，if部分应该在def on_value_change(change):中。请参阅以下代码：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib import cm
import ipywidgets as wg
%matplotlib widget

X = np.arange(5)
Y = np.arange(5)
x, y = np.meshgrid(X, Y)
rm = np.sin(x)
rm1 = np.cos(x)
rm2 = y

fig = plt.figure()
ax1 = fig.add_axes([0, 0, 1, 0.8], projection = '3d')

# Design a slider to choose which simulation to show 
slider = wg.IntSlider(value=1, min=0, max=2, description='this is slider')
slideroutput = wg.Output()
display(slider, slideroutput)

ax1.cla()
plt.title(f"This is slider number {str(slider.value)}")
ax1.plot_surface(x, y, rm1, cmap = cm.coolwarm, linewidth = 10, antialiased = False)

numberonslider = []
def on_value_change(change):
    with slideroutput:
        numberonslider.append(change['new'])
        if numberonslider[-1] == 0:
            ax1.cla()
            plt.title(f"This is slider number {str(slider.value)}")
            ax1.plot_surface(x, y, rm, cmap = cm.coolwarm, linewidth = 10, antialiased = False)
        elif numberonslider[-1] == 1:
            ax1.cla()
            plt.title(f"This is slider number {str(slider.value)}")
            ax1.plot_surface(x, y, rm1, cmap = cm.coolwarm, linewidth = 10, antialiased = False)
        elif numberonslider[-1] == 2:
            ax1.cla()
            plt.title(f"This is slider number {str(slider.value)}")
            ax1.plot_surface(x, y, rm2, cmap = cm.coolwarm, linewidth = 10, antialiased = False)
    
slider.observe(on_value_change, names='value')

输出：

如果我的解释不清楚，请再次询问。