python - Statsmodels (Patsy) 非法变量名/'Series' 对象不可调用错误

Question

更新:

该错误可能是由于我的数据集中还有一个名为“Q”的变量与 Q 函数冲突而引起的。这种情况我该如何优雅的解决呢？

<小时/>

更新:您可以下载我的数据集here .

<小时/>

我正在使用 statsmodels 和 pandas 数据帧运行简单的 OLS 回归，如下所示:

import statsmodels.formula.api as sm
import pandas as pd
df=pd.read_csv("exp.csv")
#df is a dataframe that I have containing many variable names such as AAPL, SPY, INF, etc.
for column in df: 
    result=sm.ols(formula="SPY"+" ~ "+column, data=df).fit()

但是，df 中的列名称之一是 INF。我想也许 INF 是 Pasty 的保留字，代码给出了以下错误:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/statsmodels/base/model.py", line 155, in from_formula
    missing=missing)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/statsmodels/formula/formulatools.py", line 65, in handle_formula_data
    NA_action=na_action)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/highlevel.py", line 310, in dmatrices
    NA_action, return_type)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/highlevel.py", line 165, in _do_highlevel_design
    NA_action)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/highlevel.py", line 62, in _try_incr_builders
    formula_like = ModelDesc.from_formula(formula_like)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/desc.py", line 165, in from_formula
    value = Evaluator().eval(tree, require_evalexpr=False)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/desc.py", line 400, in eval
    result = self._evaluators[key](self, tree)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/desc.py", line 221, in _eval_any_tilde
    exprs = [evaluator.eval(arg) for arg in tree.args]    
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/desc.py", line 400, in eval
    result = self._evaluators[key](self, tree)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/desc.py", line 355, in _eval_number
    "only allowed with **", tree)
patsy.PatsyError: numbers besides '0' and '1' are only allowed with **
    SPY ~ INF
          ^^^

我也尝试过使用 Q 函数:

result=sm.ols(formula="SPY"+" ~ "+"Q('INF')", data=df).fit()

但是，它给了我以下错误:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/statsmodels/base/model.py", line 155, in from_formula
    missing=missing)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/statsmodels/formula/formulatools.py", line 65, in handle_formula_data
    NA_action=na_action)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/highlevel.py", line 310, in dmatrices
    NA_action, return_type)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/highlevel.py", line 165, in _do_highlevel_design
    NA_action)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/highlevel.py", line 70, in _try_incr_builders
    NA_action)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/build.py", line 696, in design_matrix_builders
    NA_action)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/build.py", line 443, in _examine_factor_types
    value = factor.eval(factor_states[factor], data)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/eval.py", line 566, in eval
    data)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/eval.py", line 551, in _eval
    inner_namespace=inner_namespace)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/compat.py", line 36, in call_and_wrap_exc
    return f(*args, **kwargs)
  File "/home/ap248/.local/easybuild/software/2017/Core/miniconda2/4.3.27/lib/python2.7/site-packages/patsy/eval.py", line 166, in eval
    + self._namespaces))
  File "<string>", line 1, in <module>
TypeError: 'Series' object is not callable

知道如何解决吗？

Answer 1

根据此链接:http://patsy.readthedocs.io/en/latest/builtins-reference.html#patsy.builtins.Q您可以在公式中使用 Q("var") 来消除错误。

下面的代码应该可以工作。

model = sm.ols('SPY ~ Q("INF")',data=df).fit()

完整示例

import statsmodels.formula.api as sm
import pandas as pd
import numpy as np

a = np.random.rand(10,3)
df = pd.DataFrame(data=a, columns=['SPY','INF','X'])

model = sm.ols('SPY ~ Q("INF")',data=df).fit()