shakechan
2022-01-24 19:49
采纳率: 100%
浏览 36

关于Python局部变量的数据获取

问题遇到的现象和发生背景

我在学习软件包的代码过程中,想要获取到迭代过程中的变化数据,发现变量是在函数里的局部变量,在调用这个模块的时候,外部无法获取到这个变量的具体数据。

问题相关代码,请勿粘贴截图
f, (ax1, ax2, ax3) = plt.subplots(1, 3, figsize=(20,3))

simulation.plt_eps(ax=ax1, outline=False)
ax1.set_title('final permittivity distribution')

optimization.plt_objs(ax=ax2)
ax2.set_yscale('linear')

(_,_,Ez) = simulation.solve_fields()
simulation.plt_abs(ax=ax3, vmax=None)
ax1.set_title('linear field')

plt.show()

以上是外部的代码,调用optimization模块中的plt_objs

def plt_objs(self, norm=None, ax=None):
        """ Plots objective function vs. iteration"""

        iter_range = range(1, len(self.objfn_list) + 1)
        if norm == 'field':
            obj_scaled = [o/a for o, a in zip(self.objfn_list, self.src_amplitudes)]
            ax.set_ylabel('objective function / field')
        elif norm == 'power':
            obj_scaled = [o/a**2 for o, a in zip(self.objfn_list, self.src_amplitudes)]
            ax.set_ylabel('objective function / power')
        else:
            obj_scaled = self.objfn_list
            ax.set_ylabel('objective function')

        ax.plot(iter_range,  obj_scaled)
        ax.set_xlabel('iteration number')
        ax.set_title('optimization results')
        return ax

以上是optimization模块中的函数定义

我想获取到这里面obj_scaled的具体值然后导入Excel用origin画个图。

运行结果及报错内容

不管怎么加定义和参数设置,一直是未定义,无法调用

我的解答思路和尝试过的方法

在模块中增加全局变量的定义,无效

我想要达到的结果

仅仅只是想获取obj_scaled的数值,应该是一个list

2条回答 默认 最新

相关推荐 更多相似问题