问题描述
我正在尝试绘制一些符号数据。我从回归中得到了一些表达形式:
expr = '(((((((((1.0)*(2.0)))-(ER)))-(-0.37419122066665467))*0.006633039574629684)*(0.006633039574629684*((((T)-(((1.0)+(P)))))-(P))))+0.1451920626347467)'
expr
在这里是一些预测:f = f(T,P,ER)
。我知道这个特定的例子是一个疯狂的表达,但并不是很重要。基本上,假设我有一个plotdata
数据框,我试图生成具有以下内容的图:
import pandas
import sympy
import numexpr
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
expr = '(((((((((1.0)*(2.0)))-(ER)))-(-0.37419122066665467))*0.006633039574629684)*(0.006633039574629684*((((T)-(((1.0)+(P)))))-(P))))+0.1451920626347467)'
#Extract some data for surface plot but fixing one variable
plotdata = plotdata.loc[(plotdata.P == 1)]
#Extract data as lists for plotting
x = list(plotdata['T'])
y = list(plotdata['ER'])
f_real = list(plotdata['f'])
T_sympy = sympy.Symbol('T')
P_sympy = sympy.Symbol('P')
ER_sympy = sympy.Symbol('ER')
f_pred = numexpr.evaluate(expr)
fig = plt.figure()
ax = fig.add_subplot(111,projection='3d')
ax.plot_trisurf(x,y,f_real,alpha = 0.3)
ax.plot_surface(x,f_pred)
但是,我遇到f_pred
错误。
numexpr.evaluate(expr)
Traceback (most recent call last):
File "/anaconda3/lib/python3.7/site-packages/numexpr/necompiler.py",line 744,in getArguments
a = local_dict[name]
KeyError: 'ER'
During handling of the above exception,another exception occurred:
Traceback (most recent call last):
File "<ipython-input-100-c765b0f1e5ce>",line 1,in <module>
numexpr.evaluate(expr)
File "/anaconda3/lib/python3.7/site-packages/numexpr/necompiler.py",line 818,in evaluate
arguments = getArguments(names,local_dict,global_dict)
File "/anaconda3/lib/python3.7/site-packages/numexpr/necompiler.py",line 746,in getArguments
a = global_dict[name]
KeyError: 'ER'
我对numexpr
软件包不是很熟悉。但是,我一直在从1D回归到现在的3D回归。 ER
是我的一维变量,工作正常。自从一维案例以来,我显然已经对我的代码做了些微改动,但是对于为什么会出现此错误,我仍然有些茫然。
任何指针将不胜感激。
解决方法
我知道了。最后的愚蠢错误。我需要更改:
#Extract data as lists for plotting
x = list(plotdata['T'])
y = list(plotdata['ER'])
收件人:
T = list(plotdata['T'])
ER = list(plotdata['ER'])
P = list(plotdata['P'])
即numexpr.evaluate
正在寻找输入数据,而不是符号!