0

我正在尝试使用ginput在地图上注册点击，并想使用 matplotlib 小部件添加操作按钮。在下面的代码中，我可以通过将 action 的值声明为global. 如果我点击地图，action=0，如果我点击按钮，action=1，根据需要。

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

class Index:
    def test(self, event):
        global action
        action=1

# fake data
x=np.arange(30)
y=x**2
fig,ax=plt.subplots()
ax.plot(x,y)
callback = Index()
buttonname=['test']
colors=['white']
idx=[0.2]
bax,buttons={},{}

# set up list of buttons.
for i,col,button in zip(idx,colors,buttonname):
    bax[button] = plt.axes([0.92, i, 0.07, 0.07])
    buttons[button] = Button(bax[button],button,color=col,hovercolor='green')
    buttons[button].on_clicked(getattr(callback,button))

# register click on plot
while True:
     pts=plt.ginput(1)
     plt.pause(0.5)
     print("action is ",action)
     action=0 # reset

但我的困惑是，如果我采用完全相同的代码并将其放在 def 块中，则 action 的值不再传回，action始终为零。

def subtest():
    class Index:
        def test(self, event):
            global action
            action=1

    # fake data
    action=0
    x=np.arange(30)
    y=x**2
    fig,ax=plt.subplots()
    ax.plot(x,y)
    callback = Index()
    buttonname=['test']
    colors=['white']
    idx=[0.2]
    bax,buttons={},{}

    # set up list of buttons.
    for i,col,button in zip(idx,colors,buttonname):
        bax[button] = plt.axes([0.92, i, 0.07, 0.07])
        buttons[button] = Button(bax[button],button,color=col,hovercolor='green')
        buttons[button].on_clicked(getattr(callback,button))

    # register click on plot
    while True:
         pts=plt.ginput(1)
         plt.pause(0.5)
         print("action is ",action)
         action=0 # reset

res=subtest()

我很困惑为什么会发生这种情况。我尝试将类定义移到主代码中，但这没有帮助。我对任何类型的解决方案都很高兴（例如，通过参数传递动作，我不明白如何处理小部件），因为我认为使用global经常皱着眉头。但global基于 - 的解决方案也很好。

4

2 回答 2

2

actioninsidesubtest是局部的子测试，而actioninsideIndex.test是全局的。要么声明actionglobal in subtest，要么使用nonlocalin Index.test。

（我怀疑没有全局变量可能会有更好的解决方案，但由于我不熟悉 GUI 工具包，我将把它留给其他人。）

于 2021-06-30T10:05:26.407 回答
1

你想看看这篇关于闭包的文章和关于闭包的另一篇文章。我认为您需要在类和函数之外声明操作，然后在类中使用全局引用它。

我认为你不需要使用一个类——你可以在函数周围传递一个变量、布尔值、字典并实现同样的事情。

你可以使用“on_click”事件来做你想做的事，看看这个小部件教程小部件教程和这个关于 matplotlib 事件处理

这是一个使用“全局”来影响状态更改的示例代码。最好传递一个变量或使用事件来触发您希望状态生效的任何内容。

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

state = False

def change_state(btn_action):
    #do something
    btn_action= not(btn_action)
    return btn_action

def grid(val):
    global state
    #do something
    state =change_state(state)
    print(state)

    ax.grid()
    fig.canvas.draw()

    #reset value
    state =change_state(state)
    print(f'resetting state to {state}')

# fake data
x=np.arange(30)
y=x**2

#create fig, ax
fig =plt.figure()
ax= fig.subplots()
p, = ax.plot(x,y)

# set up list of buttons.

buttonname=['test']
colors=['white']
idx=[0.2]
bax,buttons={},{}

for i,col,button in zip(idx,colors,buttonname):
    bax[button] = plt.axes([0.92, i, 0.07, 0.07])
    buttons[button] = Button(bax[button],button,color=col,hovercolor='green')
    buttons[button].on_clicked(grid)

#show plot
fig.canvas.draw()
plt.show()
于 2021-06-30T12:21:12.730 回答