Question

我正在努力实现这一点：https://stackoverflow.com/a/12025554/7718153 对于matplotlib，但出了点问题。

import matplotlib.pyplot as plt

if condition1:
    q='plot'
elif condition2:
    q='logy'
elif condition3
    q='loglog'

m = globals()['plt']()
TypeError: 'module' object is not callable

plot_function = getattr(m, q) #it doens't make it to this line

我希望它能做到这一点：

if condition1:
    plt.plot(...)
elif condition2:
    plt.logy(...)
elif condition3:
    plt.loglog(...)

有谁知道我做错了什么？

编辑：对不起，我的代码输错了。现在已经解决了。

EDIT2：

这是原始来自的代码：

def plot(self):
    assert self.plotgraph == True
    plt.figure(1)
    plt.rcParams.update({'font.size': 40})
    plt.figure(figsize=(150, 70))
    plt.suptitle('alpha = '+str("{0:.2f}".format(self.alpha)))


    j=len(self.seeds)
    for k in range(9*j):
        plt.subplot(3,3*j,k+1)
        g=k%(3*j)
        if k<3*j:
            q='plot'
        elif 3*j<=k<6*j:
            q='logy'
        elif 6*j<=k:
            q='loglog'
        m = globals()['plt']()
        plot_function = getattr(m, q)
        if g<2*j:
            for i in range(j):
                if (2*i)<=g%j*2<(2*(i+1)):
                    seed_type=' seed: '+ str(i+1)
                    seed=(i+1)
        else:
            for i in range(j):
                if g%j == i:
                    seed_type=' seed: '+ str(i+1)
                    seed=(i+1)

        if g<2*j:
            if g%2==0:
                set_type=' train-set'
                plot_function(np.array(self.index),np.array(self.plotlist[seed*2+0]))
            else:
                set_type=' test-set'
                plot_function(np.array(self.index),np.array(self.plotlist[seed*2+1]))
        else:
            set_type=' train-test dist'
            plot_function(np.array(self.index),np.array(self.plotlist[seed*2+0]-np.array(self.plotlist[seed*2+1])))

        plt.grid(True)
        plt.title(q+set_type+seed_type)
    plt.tight_layout()
    plt.savefig("plot1() " +str(self.nodes[1])+' hidden nodes, alpha='+ str("{0:.2f}".format(self.alpha)) + '.png')
    plt.clf()
    plt.close()

Answer 1

m = globals()['plt']()与plt()相同。 plt是一个模块，因此无法调用。我想你想要：

m = globals()['plt']
plot_function = getattr(m, q)
plot_function()  # call this one!

据说......这种设计看起来似乎很复杂。为什么不：

if condition1:
    plt.plot()
elif condition2:
    plt.logy()
elif condition3:
    ...

如何在python中使用getattr调用函数？

1 个答案: