在进行一些 OLS 回归时,我发现它statsmodels.api.add_constant()
执行以下操作:
if _is_using_pandas(data, None) or _is_recarray(data):
from statsmodels.tsa.tsatools import add_trend
return add_trend(data, trend='c', prepend=prepend, has_constant=has_constant)
如果不是,它将被视为data
ndarray,因此您会丢失一些上下文信息(例如,作为回归变量名称的列名)。从 modin 导入 pandas 时,is_using_pandas()
上述返回False
.
可能statsmodels
需要将其添加modin
为受支持的选项,_is_using_pandas()
但现在,我想做类似的事情:
if is_using_modin_pandas(x):
from statsmodels.tsa.tsatools import add_trend
X = add_trend(x, trend='c', prepend=True, has_constant='skip')
else:
X = sm.add_constant(x)
一个人会怎么写is_using_modin_pandas()
?