python，在for循环中创建多个matplotlib按钮小部件

Question

我需要在 matplotlib 中生成一个可变数量（总数）的按钮小部件，并让每个小部件在变量 var 中存储一个不同的数字。

import matplotlib.pyplot as plt
from matplotlib.widgets import Button as BT

total = 4

class Index:
    ind = 0
        
    def plot_pick(self, event):
        var=event
        print(var)

ax_chooseplot={}
chooseplot_BT={}
callback=Index()

for i in range(0,total):
    loc=0.95-i*0.03
    ax_chooseplot[i] = plt.axes([loc-0.002, 0.87, 0.02, 0.03])
    chooseplot_BT[i] = BT(ax_chooseplot[i], '')
    chooseplot_BT[i].on_clicked(callback.plot_pick(i))         
    
plt.show()

此代码按顺序打印 i 然后按任意按钮：

TypeError: 'NoneType' object is not callable

似乎在 for 循环期间调用了函数“plot_pick”，然后丢弃了 i 的值？

我已经看到为 tkinter 按钮解决了这个问题，但我无法将这些答案应用于 matplotlib 嵌入式按钮。有没有简单的方法来解决这个问题？

Answer 1

问题来自您将i 参数传递给plot_pick 函数的方式。将其直接传递给您的函数将不起作用。一种可能的替代方法是使用functools.partial。您可以在下面找到一个示例，其中我使用functools.partial 打印被点击按钮的索引：

import matplotlib.pyplot as plt
from matplotlib.widgets import Button as BT
import functools

total = 4

class Index:
    ind = 0

    def plot_pick(self,i, event):
        print('Button '+str(i))
        

ax_chooseplot={}
chooseplot_BT={}
callback=Index()


for i in range(0,total):
    loc=0.95-i*0.03
    ax_chooseplot[i] = plt.axes([loc-0.002, 0.87, 0.02, 0.03])
    chooseplot_BT[i] = BT(ax_chooseplot[i], str(i))
    chooseplot_BT[i].on_clicked(functools.partial(callback.plot_pick,i))  
   
    
plt.show()